SSRN2026
SupervisorCorresponding author
Abstract
Cross-dataset generalization and deployment cost are two largely independent obstacles to the clinical translation of polyp segmentation. We present WaveLite-DG, a compact encoder–decoder addressing both. It combines a MobileNetV3-Large encoder, Wavelet Frequency Modules that perturb approximation-band feature statistics during training while enhancing detail-band structure, a gated context scan at the bottleneck, and a boundary-supervised decoder. We derive in closed form the energy this perturbation injects. Trained on Kvasir-SEG and CVC-ClinicDB, with CVC-ColonDB, ETIS-LaribPolypDB and CVC-300 withheld from every fitting and selection decision, WaveLite-DG reaches a three-seed mean macro unseen Dice of 0.8300 ± 0.0004, the highest of six architectures retrained under one protocol without photometric augmentation and statistically equivalent to Polyp-PVT and VM-UNet within a pre-set margin of 0.02. With colour jitter, or with the perturbation added to those two baselines, they lead by under 0.008. In every regime tested WaveLite-DG stays within 0.008 of the best model with one third or less of the parameters (7.68 M) and one fifth or less of the computation (1.93 giga multiply–accumulate operations), and it is 0.0427 more accurate than the best of four lightweight networks. Approximation-band perturbation adds +0.0651 macro Dice without photometric augmentation but only +0.0185 with colour jitter, which recovers most of the gain, and perturbing the undecomposed feature at matched energy shows no detectable difference. The contribution therefore lies in efficiency at parity rather than in a new mechanism.
Index terms
polyp segmentation, domain generalization, wavelet transform, lightweight networks.
SSRN preprint
↺ Tap or swipe to flip back