Paper Reading AI Learner

TopoLoRA-SAM: Topology-Aware Parameter-Efficient Adaptation of Foundation Segmenters for Thin-Structure and Cross-Domain Binary Semantic Segmentation

2026-01-05 17:03:45
Salim Khazem

Abstract

Foundation segmentation models such as the Segment Anything Model (SAM) exhibit strong zero-shot generalization through large-scale pretraining, but adapting them to domain-specific semantic segmentation remains challenging, particularly for thin structures (e.g., retinal vessels) and noisy modalities (e.g., SAR imagery). Full fine-tuning is computationally expensive and risks catastrophic forgetting. We propose \textbf{TopoLoRA-SAM}, a topology-aware and parameter-efficient adaptation framework for binary semantic segmentation. TopoLoRA-SAM injects Low-Rank Adaptation (LoRA) into the frozen ViT encoder, augmented with a lightweight spatial convolutional adapter and optional topology-aware supervision via differentiable clDice. We evaluate our approach on five benchmarks spanning retinal vessel segmentation (DRIVE, STARE, CHASE\_DB1), polyp segmentation (Kvasir-SEG), and SAR sea/land segmentation (SL-SSDD), comparing against U-Net, DeepLabV3+, SegFormer, and Mask2Former. TopoLoRA-SAM achieves the best retina-average Dice and the best overall average Dice across datasets, while training only \textbf{5.2\%} of model parameters ($\sim$4.9M). On the challenging CHASE\_DB1 dataset, our method substantially improves segmentation accuracy and robustness, demonstrating that topology-aware parameter-efficient adaptation can match or exceed fully fine-tuned specialist models. Code is available at : this https URL

Abstract (translated)

基于大范围预训练的分割模型(如Segment Anything Model,SAM)展现出强大的零样本泛化能力。然而,将这些基础模型调整为特定领域的语义分割任务仍然具有挑战性,尤其是在处理细长结构(例如视网膜血管)和噪声模式(例如SAR图像)时更为困难。全量微调在计算上是昂贵的,并且存在灾难性遗忘的风险。 我们提出了一种基于拓扑感知并参数高效的适应框架——TopoLoRA-SAM,用于二值语义分割任务。该方法向冻结的ViT编码器中注入低秩适配(Low-Rank Adaptation, LoRA),结合了一个轻量级的空间卷积适配器,并可选地通过不同的clDice进行拓扑感知监督。 我们在五个基准数据集上进行了评估,这些数据集涵盖了视网膜血管分割(DRIVE、STARE和CHASE_DB1)、息肉分割(Kvasir-SEG)以及SAR海陆分割(SL-SSDD),并与U-Net、DeepLabV3+、SegFormer及Mask2Former等方法进行了比较。TopoLoRA-SAM在所有数据集上获得了最高的视网膜平均Dice值和总体平均Dice值,同时仅训练了模型参数的5.2%(约490万)。 特别是在具有挑战性的CHASE_DB1数据集上,我们的方法显著提高了分割准确性和鲁棒性,证明了拓扑感知且参数高效的适应可以匹配或超过全量微调的专业模型。代码可在提供的链接处获取:this https URL

URL

https://arxiv.org/abs/2601.02273

PDF

https://arxiv.org/pdf/2601.02273.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot