Adaptive Attention Distillation for Robust Few-Shot Segmentation under Environmental Perturbations

View PDF HTML (experimental)

Abstract:Few-shot segmentation (FSS) aims to rapidly learn novel class concepts from limited examples to segment specific targets in unseen images, and has been widely applied in areas such as medical diagnosis and industrial inspection. However, existing studies largely overlook the complex environmental factors encountered in real world scenarios-such as illumination, background, and camera viewpoint-which can substantially increase the difficulty of test images. As a result, models trained under laboratory conditions often fall short of practical deployment requirements. To bridge this gap, in this paper, an environment-robust FSS setting is introduced that explicitly incorporates challenging test cases arising from complex environments-such as motion blur, small objects, and camouflaged targets-to enhance model's robustness under realistic, dynamic conditions. An environment robust FSS benchmark (ER-FSS) is established, covering eight datasets across multiple real world scenarios. In addition, an Adaptive Attention Distillation (AAD) method is proposed, which repeatedly contrasts and distills key shared semantics between known (support) and unknown (query) images to derive class-specific attention for novel categories. This strengthens the model's ability to focus on the correct targets in complex environments, thereby improving environmental robustness. Comparative experiments show that AAD improves mIoU by 3.3% - 8.5% across all datasets and settings, demonstrating superior performance and strong generalization. The source code and dataset are available at: this https URL.

Comments: 12 pages, 5 figures

Subjects:

Computer Vision and Pattern Recognition (cs.CV)

Cite as: arXiv:2601.03596 [cs.CV]

(or arXiv:2601.03596v3 [cs.CV] for this version)

https://doi.org/10.48550/arXiv.2601.03596

arXiv-issued DOI via DataCite

Submission history

From: Qianyu Guo [view email] [v1] Wed, 7 Jan 2026 05:27:12 UTC (2,461 KB) [v2] Fri, 6 Feb 2026 01:25:07 UTC (2,578 KB) [v3] Sun, 29 Mar 2026 15:38:05 UTC (1,841 KB)