Reproduce “Leveraging Complementary Attention maps in vision transformers for OCT image analysis” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Haz Sameen Shahgir, Tanjeem Azwad Zaman, Khondker Salman Sayeed, Md. Asif Haider, Sheikh Saifur Rahman Jony, M. Sohel Rahman

Researchers at Bangladesh University of Engineering and Technology developed an automated pipeline for detecting multiple ophthalmic biomarkers from OCT scans by integrating complementary Vision Transformer architectures and knowledge distillation. Their MaxViT-EVA02 ensemble achieved a patient-wise F1 score of 0.8527 on an unseen test set, securing first place in the IEEE Video and Image Processing Cup 2023, while a distilled MaxViT model maintained similar performance with significantly reduced computational cost.

Key insights

  • Different vision architectures are uniquely suited for detecting OCT biomarkers based on their spatial characteristics: MaxViT excels at local features, while EVA-02 demonstrates stronger performance on global features.
  • Combining models with complementary strengths (specifically MaxViT and EVA-02) through ensembling significantly improves overall biomarker detection accuracy.
  • Knowledge distillation from a high-performing ensemble can produce a single, computationally efficient model that maintains or slightly surpasses the ensemble's accuracy, making it practical for clinical deployment.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.