Annotation-Free Representation Learning for Polyp Retrieval in Colonoscopy Videos
Organizations: Department of Mathematics, University of Padova, Padova, Italy · Cosmo Intelligent Medical Devices, Dublin, Ireland · LTCI, Telecom Paris, Institut Polytechnique de Paris, Palaiseau, France
Abstract
Colonoscopy is inherently comparative: reliable visual assessment of polyps often depends on complementary views acquired under different viewpoints and imaging conditions. Retrieving such views requires polyp tracklet representations, which recent contrastive methods learn from temporal cues but still build from manually annotated bounding boxes. We propose an annotation-free framework that constructs tracklet views directly from raw colonoscopy videos, enabling temporal self-supervised learning on full-procedure videos, including frames without visible polyps. On REAL-Colon, our method achieves state-of-the-art results on tracklet retrieval and re-identification, outperforming both self-supervised and supervised baselines. We then introduce few-shot polyp retrieval, a task closer to clinical use, where a few observations of a target polyp are used to retrieve its remaining tracklets: with only two examples, our method surpasses most competing methods given ten. Finally, we show that a modest number of unlabeled procedures suffices for competitive performance, making on-site model development more practical. Code is available at https://github.com/lparolari/freerep.
Figures & tables
| Method | Tracklet Retrieval | Tracklet ReID | Few-Shot Polyp Retrieval | ||||||||
| Params | FLOPs | mAP | HR@1 | AUROC | AUPR | mAP ( ) | |||||
| (M) | (G) | ||||||||||
| General Purpose FM | |||||||||||
| DinoV2-S/14 [ 12 ] | 22.1 | 44.2 | 47.31 | 89.81 | 80.07 | 29.77 | 47.91 | 57.04 | 62.73 | 65.64 | 72.57 |
| DinoV2-g/14 [ 12 ] | 1136 | 2331 | 48.29 | 90.35 | 78.23 | 27.04 | 48.08 | 57.20 | 62.40 | 66.46 | 72.13 |
| DinoV3-S/16 [ 13 ] | 21.6 | 34.7 | 48.95 | 88.99 | 80.47 | 31.77 | 47.60 | 57.36 | 62.95 | 66.80 | 76.59 |
| Training setup | Tracklet Retrieval | Tracklet ReID | |||||
| Annot. | Loss | Views | Full | mAP | HR@1 | AUROC | AUPR |
| ID+Box | SupCon | BBox | ✗ | 57.52 | 84.24 | 94.33 | 43.81 |
| Box | Noise-Aw. | BBox | ✗ | 61.49 | 91.17 | 93.89 | 51.78 |
| Stats | Noise-Aw. | Synth. | ✓ | 65.60 | 91.71 | 95.14 | 57.72 |
| – | Noise-Aw. | Frame | ✓ | 52.27 | 84.10 | 89.24 | 43.74 |
| – | SupCon | Crop | ✓ | 64.31 | 89.54 | 94.97 | 49.83 |