Search papers, labs, and topics across Lattice.
This study addresses the limitations of traditional ultrasound-based models for preterm birth (PTB) prediction by framing the problem as multiple instance learning (MIL), where each patient is represented by a collection of transvaginal ultrasound (TVUS) images. The authors introduce a novel Gaussian Mixture Model (GMM) pooling technique that captures the feature distribution of all images in a bag, rather than reducing them to a single point estimate, thereby accounting for intra-patient variability. Evaluation on a private clinical cohort shows a significant improvement in PTB prediction performance, with the model's PR-AUC increasing from 0.44 to 0.56, while also achieving state-of-the-art results on a lymph node metastasis benchmark.
GMM pooling not only enhances preterm birth prediction accuracy but also sets a new standard for image-based classification tasks by effectively modeling intra-patient variability.
Preterm birth (PTB) prediction can enable targeted surveillance and timely intervention, yet most ultrasound-based models use a single selected transvaginal ultrasound (TVUS) frame per patient despite routine exams acquiring multiple cervical images. We formulate PTB prediction as a multiple instance learning (MIL) problem, representing each patient as a variable-sized bag of TVUS images with a single outcome label. To move beyond standard MIL aggregators that collapse a bag into a point estimate, we propose a Gaussian Mixture Model (GMM) pooling, which summarizes all images in a bag into a fixed-length representation by modeling their feature distribution. This design captures intra-patient variability. We evaluate the method on a private clinical cohort and on a public lymph node metastasis benchmark. For PTB prediction, GMM pooling improves over the instance-based model PR-AUC from 0.44 to 0.56. On the lymph node benchmark, it achieves state-of-the-art performance with 0.91 F1-score and 0.89 ROC-AUC for classification and 0.18 MAE for regression. The code is publicly available at https://github.com/HussainAlasmawi/GMM_Pooling.
