Token-Based Dual-view Fusion and Adaptation of Large Vision Models for Breast Cancer Classification
Summary
A token-centric dual-view learning framework is proposed for accurate breast cancer classification from mammography, integrating craniocaudal (CC) and mediolateral oblique (MLO) views. This framework unifies prompt-based adaptation and cross-view fusion within a frozen vision transformer backbone. It reformulates inter-view interaction as structured token-level communication, where dedicated fusion tokens explicitly encode bidirectional information exchange via cross-attention. Fusion modules are inserted at multiple transformer depths, enabling progressive interaction and hierarchical propagation of complementary information. Experiments on VinDr-Mammo and CMMD datasets show consistent improvements over baselines. On VinDr-Mammo BI-RADS classification, it achieved a 50.40% F1-score and 0.8090 AUC, including a 0.10 AUC improvement over a dual-view fusion baseline in the binary setting.
Key takeaway
For Machine Learning Engineers developing multi-view medical image analysis models, this token-centric fusion approach offers a robust method to integrate complementary information. You should consider implementing multi-depth token-level communication via cross-attention within your vision transformer backbones to improve classification accuracy, especially when dealing with complex diagnostic tasks like breast cancer detection from mammograms. This can lead to more effective hierarchical information propagation and better diagnostic performance.
Key insights
A token-centric, multi-depth fusion framework enhances breast cancer classification by integrating complementary mammography views.
Principles
- Reformulate inter-view interaction as structured token-level communication.
- Enable progressive, repeated interaction across the encoder hierarchy.
Method
Dedicated fusion tokens explicitly encode bidirectional information exchange between views via cross-attention, inserted at multiple transformer depths, then reintegrated and refined by subsequent layers.
In practice
- Apply token-based fusion for integrating complementary medical imaging views.
- Utilize multi-depth interaction to propagate hierarchical information effectively.
Topics
- Breast Cancer Classification
- Mammography
- Vision Transformers
- Multi-view Learning
- Token-based Fusion
- Medical Imaging
Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.