CAFE-Net: Cross-Attention and Feature Exploration Network for polyp segmentation
Colorectal polyp segmentation can help physicians screen colonoscopy images, which is essential for preventing colorectal cancer. The segmentation of polyps encounters multiple challenges, like small size, uneven brightness, blurred edges, and the potential confusion of folds with objects. Although existing methods have shown promising performance in addressing these challenges, there are three main shortcomings: (1) during the segmentation process, small polyp objects are lost, (2) the decoder stage faces challenges in restoring fine-grained details of the features, and (3) the limited capacity to aggregate multi-scale features. We propose a cross-attention and feature exploration network (CAFE-Net) for polyp segmentation to address these challenges. The work offers the following contributions: (1) a feature supplement and exploration module (FSEM) supplements in missing details and explores latent features, (2) a cross-attention decoder module (CADM) effectively preserves features from lower layers and restores fine-grained information, and (3) a multi-scale feature aggregation (MFA) module maximizes the utilization of previously learned features. We conducted extensive experiments and compared CAFE-Net with nine state-of-the-art (SOTA) methods. The CAFE-Net demonstrates the best segmentation accuracy across multiple polyp datasets as well as has an obvious advantage in segmenting small polyp objects.
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
CAFE-Net: Cross-Attention and Feature Exploration Network for polyp segmentation
Colorectal polyp segmentation can help physicians screen colonoscopy images, which is essential for preventing colorectal cancer. The segmentation of polyps encounters multiple challenges, like small size, uneven brightn…
CafeMed: Causal Attention Fusion Enhanced Medication Recommendation
Medication recommendation systems play a crucial role in assisting clinicians with personalized treatment decisions. While existing approaches have made significant progress in learning medication representations, they s…
Recommendation SystemsCausal Interpretation of Sparse Autoencoder Features in Vision
Understanding what sparse auto-encoder (SAE) features in vision transformers truly represent is usually done by inspecting the patches where a feature's activation is highest. However, self-attention mixes information ac…
Causally-Guided Automated Feature Engineering with Multi-Agent Reinforcement Learning
Automated feature engineering (AFE) enables AI systems to autonomously construct high-utility representations from raw tabular data. However, existing AFE methods rely on statistical heuristics, yielding brittle features…
Multi-agent Reinforcement LearningFeature EngineeringCAFE-AD: Cross-Scenario Adaptive Feature Enhancement for Trajectory Planning in Autonomous Driving
Imitation learning based planning tasks on the nuPlan dataset have gained great interest due to their potential to generate human-like driving behaviors. However, open-loop training on the nuPlan dataset tends to cause c…
Autonomous DrivingFeature ImportanceImitation LearningTrajectory Planning