A Novel Momentum-Based Deep Learning Techniques for Medical Image Classification and Segmentation


Biswas K., Pal R., Patel S., Jha D., Karri M., Reza A., ...Daha Fazla

15th International Workshop on Machine Learning in Medical Imaging, MLMI 2024 was held in conjunction with the 27th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2024, Marrakush, Fas, 06 Ekim 2024, cilt.15241 LNCS, ss.1-11, (Tam Metin Bildiri) identifier

  • Yayın Türü: Bildiri / Tam Metin Bildiri
  • Cilt numarası: 15241 LNCS
  • Doi Numarası: 10.1007/978-3-031-73284-3_1
  • Basıldığı Şehir: Marrakush
  • Basıldığı Ülke: Fas
  • Sayfa Sayıları: ss.1-11
  • Anahtar Kelimeler: Liver segmentation, Lung Cancer Segmentation, Medical Image Classification, Polyp Segmentation
  • İstanbul Üniversitesi Adresli: Evet

Özet

Accurately segmenting different organs from medical images is a critical prerequisite for computer-assisted diagnosis and intervention planning. This study proposes a deep learning-based approach for segmenting various organs from CT and MRI scans and classifying diseases. Our study introduces a novel technique integrating momentum within residual blocks for enhanced training dynamics in medical image analysis. We applied our method in two distinct tasks: segmenting liver, lung, & colon data and classifying abdominal pelvic CT and MRI scans. The proposed approach has shown promising results, outperforming state of-the-art methods on publicly available benchmarking datasets. For instance, in the lung cancer segmentation dataset, our approach yielded significant enhancements over the TransNetR model, including a 5.72% increase in dice score, a 5.04% improvement in mean Intersection over Union (mIoU), an 8.02% improvement in recall, and a 4.42% improvement in precision. Hence, incorporating momentum led to state-of-the art performance in both segmentation and classification tasks, representing a significant advancement in the field of medical imaging. The code is available at https://github.com/koushik313/momentum.