mlsanityData to 14 Oct 2026

End-to-end Training for Whole Image Breast Cancer Diagnosis using An All Convolutional Design

arXiv 1711.05775 ↗First posted 15 Nov 2017cs.CVcited by 88

Not on the map. Why?

Abstract (arXiv)

We develop an end-to-end training algorithm for whole-image breast cancer diagnosis based on mammograms. It requires lesion annotations only at the first stage of training. After that, a whole image classifier can be trained using only image level labels. This greatly reduced the reliance on lesion annotations. Our approach is implemented using an all convolutional design that is simple yet provides superior performance in comparison with the previous methods. On DDSM, our best single-model achieves a per-image AUC score of 0.88 and three-model averaging increases the score to 0.91. On INbreast, our best single-model achieves a per-image AUC score of 0.96. Using DDSM as benchmark, our models compare favorably with the current state-of-the-art. We also demonstrate that a whole image model trained on DDSM can be easily transferred to INbreast without using its lesion annotations and using only a small amount of training data. Code availability: https://github.com/lishen/end2end-all-conv

Key facts. Cited by 88 papers in the map (data through 14 Oct 2026). First posted on arXiv 15 Nov 2017.

Builds on

No reference list yet.

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2021: 25; 2022: 21; 2023: 13; 2024: 7; 2025: 13; 2026: 9.

88 papers.

Sort:

By area:

  1. 2021Do Feature Attribution Methods Correctly Attribute Features?cited by 98
  2. 2022Selection by Prediction with Conformal p-valuescited by 69
  3. 2021Memory-aware curriculum federated learning for breast cancer classificationcited by 20
  4. 2021Machine Learning Towards Intelligent Systems: Applications, Challenges, and Opportunitiescited by 14
  5. 2021Breast Cancer Diagnosis in Two-View Mammography Using End-to-End Trained EfficientNet-Based Convolutional Networkcited by 13
  6. 2021Detection and severity classification of COVID-19 in CT images using deep learningcited by 12
  7. 2022Multi-view Local Co-occurrence and Global Consistency Learning Improve Mammogram Classification Generalisationcited by 12
  8. 2022Domain generalization in deep learning-based mass detection in mammography: A large-scale multi-center studycited by 10
  9. 2022Meta Ordinal Regression Forest for Medical Image Classification with Ordinal Labelscited by 7
  10. 2022Multi-View Hypercomplex Learning for Breast Cancer Screeningcited by 7
  11. 2023MammoDG: Generalisable Deep Learning Breaks the Limits of Cross-Domain Multi-Center Breast Cancer Screeningcited by 7
  12. 2021MEDUSA: Multi-scale Encoder-Decoder Self-Attention Deep Neural Network Architecture for Medical Image Analysiscited by 6
38 more
  1. 2022Three Applications of Conformal Prediction for Rating Breast Density in Mammographycited by 6
  2. 2023Attention-Map Augmentation for Hypercomplex Breast Cancer Classificationcited by 6
  3. 2023Generalizing Medical Image Representations via Quaternion Wavelet Networkscited by 6
  4. 2025Efficient Breast and Ovarian Cancer Classification via ViT-Based Preprocessing and Transfer Learningcited by 6
  5. 2021Beyond Fine-tuning: Classifying High Resolution Mammograms using Function-Preserving Transformationscited by 5
  6. 2022Breast Cancer Classification using Deep Learned Features Boosted with Handcrafted Featurescited by 5
  7. 2022Estimating and Controlling for Equalized Odds via Sensitive Attribute Predictorscited by 5
  8. 2022Learning to diagnose common thorax diseases on chest radiographs from radiology reports in Vietnamesecited by 5
  9. 2023Detection of Mild Cognitive Impairment Using Facial Features in Video Conversationscited by 5
  10. 2024Decoupling Feature Extraction and Classification Layers for Calibrated Neural Networkscited by 5
  11. 2021Meta-repository of screening mammography classifierscited by 4
  12. 2021Multi-task fusion for improving mammography screening data classificationcited by 4
  13. 2022What is Your Metric Telling You? Evaluating Classifier Calibration under Context-Specific Definitions of Reliabilitycited by 4
  14. 2023Leveraging Transformers to Improve Breast Cancer Classification and Risk Assessment with Multi-modal and Longitudinal Datacited by 4
  15. 2021Low-Regret Active learningcited by 3
  16. 2021A Real Use Case of Semi-Supervised Learning for Mammogram Classification in a Local Clinic of Costa Ricacited by 3
  17. 2022Calibrating Histopathology Image Classifiers using Label Smoothingcited by 3
  18. 2023On the Feasibility of Machine Learning Augmented Magnetic Resonance for Point-of-Care Identification of Diseasecited by 3
  19. 2023MDF-Net for abnormality detection by fusing X-rays with clinical datacited by 3
  20. 2023The effect of data augmentation and 3D-CNN depth on Alzheimer's Disease detectioncited by 3
  21. 2024Prototype-based Interpretable Breast Cancer Prediction Models: Analysis and Challengescited by 3
  22. 2025Augmented Intelligence for Multimodal Virtual Biopsy in Breast Cancer Using Generative Artificial Intelligencecited by 3
  23. 2025Balancing Two Classifiers via A Simplex ETF Structure for Model Calibrationcited by 3
  24. 2021Invertible Residual Network with Regularization for Effective Medical Image Segmentationcited by 2
  25. 2021Deep Curriculum Learning in Task Space for Multi-Class Based Mammography Diagnosiscited by 2
  26. 2021Gradient Frequency Modulation for Visually Explaining Video Understanding Modelscited by 2
  27. 2022Towards Clinical Practice: Design and Implementation of Convolutional Neural Network-Based Assistive Diagnosis System for COVID-19 Case Detection from Chest X-Ray Imagescited by 2
  28. 2023On the Within-Group Fairness of Screening Classifierscited by 2
  29. 2025Dynamic Residual Encoding with Slide-Level Contrastive Learning for End-to-End Whole Slide Image Representationcited by 2
  30. 2021Interpretable COVID-19 Chest X-Ray Classification via Orthogonality Constraintcited by 1
  31. 2021Research on the Detection Method of Breast Cancer Deep Convolutional Neural Network Based on Computer Aidcited by 1
  32. 2021An XAI Approach to Deep Learning Models in the Detection of DCIScited by 1
  33. 2021Multimodal Breast Lesion Classification Using Cross-Attention Deep Networkscited by 1
  34. 2021Unsupervised multi-latent space reinforcement learning framework for video summarization in ultrasound imagingcited by 1
  35. 2021A Teacher-Student Markov Decision Process-based Framework for Online Correctional Learningcited by 1
  36. 2022Machine vision for vial positioning detection toward the safe automation of material synthesiscited by 1
  37. 2022Deep Learning Hyperparameter Optimization for Breast Mass Detection in Mammogramscited by 1
  38. 2022Slice-level Detection of Intracranial Hemorrhage on CT Using Deep Descriptors of Adjacent Slicescited by 1

50 most cited of 88 All 88 (JSON)