LHU-VT: A Lightweight Hypercomplex U-Net with Vessel Thickness-Guided Dice Loss for retinal vessel segmentation

Vision loss is often caused by retinal disorders, such as age-related macular degeneration and diabetic retinopathy, where early indicators like microaneurysms and hemorrhages appear as changes in retinal blood vessels. Accurate segmentation of these vessels in retinal images is essential for early diagnosis. However, retinal vessel segmentation presents challenges due to complex vessel structures, low contrast, and dense branching patterns, which are further complicated in resource-limited settings requiring lightweight solutions. To address these challenges, we propose a novel Lightweight Hypercomplex U-Net¹ (LHUN) with Vessel Thickness-Guided Dice Loss (VTDL), collectively called LHU-VT. LHUN utilizes hypercomplex octonions to capture intricate patterns and cross-channel relationships in fundus images, reducing parameter count and enabling edge deployment. The VTDL component applies vessel thickness-guided weights to address class imbalance, thereby enhancing segmentation accuracy. Our experiments show that LHU-VT significantly outperforms current methods, achieving up to 2.4× fewer FLOPs, 4.4× fewer parameters, and 2.6× smaller model size. The model achieves AUC scores of 0.9938, 0.9879, 0.9988, and 0.9808, respectively, on four benchmark datasets CHASE, DRIVE, STARE, and HRF.