OraRefer: Segmentation-Guided Multi-Photograph Fusion for Oral Cancer Screening with Guaranteed Patient-Level Sensitivity
Organizations: Department of Medical Science and Technology Indian Institute of Technology Kharagpur, India · Dr. B. C. Roy Multi-Speciality Medical Research Centre Indian Institute of Technology Kharagpur, India
Abstract
Oral cancer and the oral potentially malignant disorders (OPMD) that can precede it are visible on the oral mucosa, which makes smartphone photographs a promising tool for screening. Deep learning models for this task classify individual photographs. However, screening refers patients, not photographs: a patient is usually photographed several times, and the referral decision must miss no more than a set share of patients with disease. We propose OraRefer, a patient-level screening model that reads each photograph with a segmentation-guided foundation model that conditions its class decision on the predicted lesion, combines all photographs of a patient with risk factors, and refers through a threshold with a finite-sample sensitivity guarantee. The model localizes the mouth itself, so inference needs no annotation. On 3000 photographs of 714 patients, patient-level fusion reaches an AUROC of 0.919 and outperforms attention-based multiple-instance learning. At a 90% sensitivity target, OraRefer detects 95% of patients with OPMD or cancer and misses 3 of 61 patients with cancer, at a specificity of 0.610. Because the threshold is calibrated without retraining, each clinic can set its own guaranteed operating point on a local calibration sample.
Figures & tables
| Macro AUROC | [95% interval] | Folds lower | Cancer AUROC | Macro-F1 | Lesion Dice | |
| OraRefer | 0.893 | – | – | 0.942 | 0.704 | 0.580 |
| w/o segmentation guidance | 0.869 | 0.023 [ 0.033, 0.015] | 5 | 0.923 | 0.688 | – |
| w/o lesion-guided pooling | 0.884 | 0.009 [ 0.016, 0.002] | 4 | 0.919 | 0.703 | 0.587 |
| w/o region-affinity attention | 0.893 | 0.001 [ 0.005, 0.007] | 1 | 0.943 | 0.703 | 0.561 |
| w/o supervised contrastive loss | 0.885 | 0.007 [ 0.014, 0.002] | 4 | 0.934 | 0.705 | 0.581 |
| with annotated mouth box | 0.891 | 0.001 [ 0.003, 0.000] | 3 | 0.940 | 0.707 | 0.583 |
| AUROC | Sens. | Spec. | Ref. | |
|---|---|---|---|---|
| Risk factors only | 0.760 [0.724, 0.795] | 0.958 | 0.201 | 88% |
| Image only | 0.909 [0.888, 0.930] | 0.958 | 0.562 | 72% |
| Max of | 0.878 [0.853, 0.903] | 0.950 | 0.538 | 72% |
| Mean of | 0.889 [0.865, 0.914] | 0.958 | 0.502 | 74% |
| Noisy-OR | 0.851 [0.824, 0.879] | 0.961 | 0.387 | 80% |
| ABMIL | 0.890 [0.867, 0.914] | 0.953 | 0.508 | 74% |
Oncology
Oncology studies cancer biology, prevention, diagnosis, treatment, and data-driven ways to improve patient care.
See all 264 papers