MEGANet-W: A Wavelet-Driven Edge-Guided Attention Framework for Weak Boundary Polyp Detection
Journal:
arXiv
Published Date:
Jul 3, 2025
Abstract
Colorectal polyp segmentation is critical for early detection of colorectal
cancer, yet weak and low contrast boundaries significantly limit automated
accuracy. Existing deep models either blur fine edge details or rely on
handcrafted filters that perform poorly under variable imaging conditions. We
propose MEGANet-W, a Wavelet Driven Edge Guided Attention Network that injects
directional, parameter free Haar wavelet edge maps into each decoder stage to
recalibrate semantic features. Our two main contributions are: (1) a two-level
Haar wavelet head for multi orientation edge extraction; and (2) Wavelet Edge
Guided Attention (WEGA) modules that fuse wavelet cues with reverse and input
branches. On five public polyp datasets, MEGANetW consistently outperforms
existing methods, improving mIoU by up to 2.3% and mDice by 1.2%, while
introducing no additional learnable parameters.