English

VAMAE: Vessel-Aware Masked Autoencoders for OCT Angiography

Computer Vision and Pattern Recognition 2026-04-09 v1

Abstract

Optical coherence tomography angiography (OCTA) provides non-invasive visualization of retinal microvasculature, but learning robust representations remains challenging due to sparse vessel structures and strong topological constraints. Many existing self-supervised learning approaches, including masked autoencoders, are primarily designed for dense natural images and rely on uniform masking and pixel-level reconstruction, which may inadequately capture vascular geometry. We propose VAMAE, a vessel-aware masked autoencoding framework for self-supervised pretraining on OCTA images. The approach incorporates anatomically informed masking that emphasizes vessel-rich regions using vesselness and skeleton-based cues, encouraging the model to focus on vascular connectivity and branching patterns. In addition, the pretraining objective includes reconstructing multiple complementary targets, enabling the model to capture appearance, structural, and topological information. We evaluate the proposed pretraining strategy on the OCTA-500 benchmark for several vessel segmentation tasks under varying levels of supervision. The results indicate that vessel-aware masking and multi-target reconstruction provide consistent improvements over standard masked autoencoding baselines, particularly in limited-label settings, suggesting the potential of geometry-aware self-supervised learning for OCTA analysis.

Keywords

Cite

@article{arxiv.2604.06583,
  title  = {VAMAE: Vessel-Aware Masked Autoencoders for OCT Angiography},
  author = {Ilerioluwakiiye Abolade and Prince Mireku and Kelechi Chibundu and Peace Ododo and Emmanuel Idoko and Promise Omoigui and Solomon Odelola},
  journal= {arXiv preprint arXiv:2604.06583},
  year   = {2026}
}

Comments

8 pages, 5 figures. Accepted at ICPR 2026

R2 v1 2026-07-01T11:58:31.181Z