A 9-class differential diagnosis system for severe central nervous system infections, with primary amoebic meningoencephalitis (PAM) caused by Naegleria fowleri as the highest-stakes target class. Built in PyTorch on the RigoBERTa Clinical encoder, with a 5-model deep ensemble, dual-gate out-of-distribution detection (Mahalanobis plus logit-energy), and Mondrian per-class conformal prediction so coverage is controlled within each diagnostic class.
- Dual-gate OOD detection combining Mahalanobis distance with a logit-energy criterion
- Mondrian per-class conformal abstention so the model declines rather than guess on out-of-domain cases
- 5-model deep ensemble over RigoBERTa Clinical for the 9-class differential