Token-Based Dual-view Fusion and Adaptation of Large Vision Models for Breast Cancer Classification

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Health & Medical Research · Depth: Expert, quick

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

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

Topics

Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.