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

· Source: Takara TLDR - Daily AI Papers · Field: Health & Wellbeing — Health & Medical Research, Medical Devices & Health Technology · Depth: Expert, medium

Summary

A novel token-centric dual-view learning framework is introduced for accurate breast cancer classification from mammography, integrating complementary craniocaudal (CC) and mediolateral oblique (MLO) views. This framework unifies prompt-based adaptation and cross-view fusion within a frozen vision transformer backbone, addressing limitations of existing multi-view approaches. It reformulates inter-view interaction using dedicated "fusion tokens" that explicitly encode bidirectional information exchange via cross-attention, acting as intermediate carriers of cross-view dependencies. Unlike conventional single-layer fusion, these fusion modules are inserted at multiple transformer depths, enabling progressive and repeated interaction across the encoder hierarchy. Experiments on VinDr-Mammo and CMMD datasets show consistent improvements, achieving a 50.40% F1-score and 0.8090 AUC on VinDr-Mammo BI-RADS classification, including a 0.10 AUC improvement over a dual-view fusion baseline in the binary setting.

Key takeaway

For Computer Vision Engineers developing medical imaging AI, particularly for breast cancer classification, you should consider implementing token-based dual-view fusion. This approach, which uses dedicated fusion tokens for progressive, multi-depth interaction between mammogram views, significantly improves classification accuracy. You can achieve superior performance, as demonstrated by a 0.10 AUC improvement, by moving beyond single-stage feature fusion and leveraging structured token-level communication within your vision transformer backbones.

Key insights

Token-based fusion and multi-depth interaction in vision transformers enhance breast cancer classification from dual-view mammograms.

Principles

Method

A token-centric framework unifies prompt-based adaptation and cross-view fusion in a frozen vision transformer. Fusion tokens facilitate bidirectional interaction across multiple transformer depths, then reintegrate for hierarchical propagation.

In practice

Topics

Code references

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

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.