SWIFT-Cell: Segmentation Without Iterative Flow Tracking
Organizations: Turocrates.ai, Mumbai, India · Department of Pathology, ESIC Medical College and Hospital, Kalaburagi, Karnataka, India
Abstract
Generalist segmenters such as Cellpose-SAM (CPSAM) outline cells accurately across stains but slowly; watershed detectors such as QuPath's are fast but less accurate. In time-critical use, such as interactive retrieval of cells from a user-drawn region, CPSAM at the reference settings takes 58-303 s per measured 8192 x 8192 region on an RTX 4500 Ada GPU, beyond a seconds-scale interactive target. We distil CPSAM into SWIFT-Cell (SWIFT), a 0.63M-parameter U-Net trained only on CPSAM masks of 1,202 H&E and IHC tiles. On three 8192 x 8192 development regions SWIFT takes 2.0-5.2 s on 4 CPU cores in int8, 2.5-5.1x faster than QuPath's detector on the same cores, and 0.36-0.49 s on a GPU. Against human annotations on eight H&E, IHC and ISH sets it keeps 95% of CPSAM's panoptic quality (PQ) on PanNuke (0.593 vs. 0.627) and is level on NuCLS (0.546 vs. 0.540). On Ki-67 IHC masks scored to the outline centre it is 0.205 PQ above QuPath's better mode, 0.079 below CPSAM and 0.027 below InstanSeg (CI includes 0; at the outer edge 0.008 below CPSAM); on 212 pathologist-outlined nuclei in Ki-67 fields of 12 untouched patients its recall is 0.925 (CPSAM 0.920, InstanSeg 0.901; paired difference CIs include 0); one method-blinded pathologist accepted its outlines in 96% of 100 fields (60 from two development tissue patients, 40 hematology; CPSAM 83%, QuPath 53%; tissue 100% for SWIFT and CPSAM); on ER/PR points from 12 patients its detection F1 is 0.882 (CPSAM 0.896, InstanSeg 0.857), and on 4-reader PD-L1 points 0.83 (CPSAM 0.85, readers 0.87). SWIFT is close to CPSAM in accuracy and faster than watershed detection.
Figures & tables
| Method (hardware) | Time (s) | F1 vs. CPSAM | DAB + |
|---|---|---|---|
| H&E / Ki-67 / CD3 | H&E / Ki-67 / CD3 | area | |
| CPSAM 0.30 µm/px (GPU) | 86 / 303 / 58 | 1 / 1 / 1 | 1.00 |
| CPSAM 0.5 µm/px (GPU) | 34 / 131 / 23 | .93 / .95 / .84 | 1.00 |
| QuPath (4 CPU thr.) | 10.2 / 26.6 / 5.1 | .68 / .54 / .61 | 0.68 |
| InstanSeg [ 4 ] (GPU fp16) | 1.73 / 5.95 / 1.19 | .87 / .84 / .76 | 0.85 |
| InstanSeg (4 CPU thr.) | 78 / 199 / 49 | .88 / .84 / .76 | 0.85 |
| Set | Stain, tissue; reference | Patients | Nuclei | Metric | SWIFT | CPSAM | InstanSeg | StarDist | QuPath | vs. CPSAM [95% CI] |
|---|---|---|---|---|---|---|---|---|---|---|
| PanNuke fold 1 [ 16 ] | H&E, 19 tissues; masks | no ids | 63,218 | PQ | .593 | .627 | .589 | .577 | .417 | .033 [ .036, .030] |
| NuCLS [ 17 ] | H&E, breast; polygons | 122 | 19,175 | PQ | .546 | .540 | .524 | .527 | .413 | .006 [.000, .011] |
| MoNuSeg test [ 18 ] | H&E, 7 organs; masks | 14 | 6,697 | PQ | .629 | .656 | .657 | .665 | .503 | .027 [ .039, .010] |
| BC-DeepLIIF [ 19 ] | Ki-67, breast; tumour masks | unknown (10–15 inferred) | 6,240 | PQ | .609 | .688 | .637 | .496 | .344 (.405) | .079 [ .089, .064] |
| EndoNuke [ 20 ] | ER/PR, endometrium; points | 12 | 210,419 | F1 | .882 | .896 | .857 | .684 | .803 | .014 [ .019, .005] |
| IGNITE, 4 readers [ 14 ] | PD-L1, lung; points | 10 | 9,676 | F1 | .826 | .851 | n.m. | n.m. | .657 | .025 [ .062, .005] |
| Decoder | F1 vs. CPSAM | PQ | Flips (%) |
|---|---|---|---|
| H&E/Ki-67/CD3 | PanNuke/NuCLS | H&E/Ki-67/CD3 | |
| Offsets, output | .886/.879/.728 | .593/.546 | 6.0/25.5/7.2 |
| Offsets, 0.5 µm | .886/.876/.727 | .587/.544 | 20.2/34.8/21.5 |
| Heatmap WS, output | .882/.864/.726 | .590/.544 | 8.0/36.3/8.4 |
| Nearest seed | .872/.842/.719 | .571/.535 | 18.5/53.3/17.8 |
Histopathology Image Segmentation
Histopathology image segmentation identifies tissue regions, cells, and their boundaries in microscopic pathology images.
See all 53 papers