AI Listens to the Womb: Deep Learning Model Predicts Preterm Birth from Uterine Electrical Signals

Every year, an estimated 15 million babies around the world are born preterm, and complications of prematurity remain a leading cause of death among newborns. For clinicians, the central challenge has always been anticipation: identifying, weeks in advance, which pregnancies will end too soon. Now a team of researchers in China has unveiled a deep learning system that reads the electrical chatter of the pregnant uterus and converts it into a personalized risk score, achieving strikingly high accuracy in distinguishing women who will deliver early from those who will carry to term. The study, published in Medical & Biological Engineering & Computing, describes a model called CWT-AuxNet, which reached an area under the receiver operating characteristic curve, or AUC, of 0.932 at the level of individual patients, a figure that places it among the strongest noninvasive preterm birth predictors reported to date.

The signal at the heart of the work is the electrohysterogram, or EHG, a recording of the electrical activity that coordinates contractions of the uterine muscle. Electrodes placed on the abdominal wall pick up these faint potentials noninvasively, much as an electrocardiogram captures the heart’s rhythm. Researchers have known for decades that the EHG changes character as pregnancy progresses: as labor approaches, the electrical bursts that sweep across the uterus shift toward lower frequencies and become more synchronized, a physiological signature of the muscle preparing for coordinated contractions. What has been harder is turning that knowledge into a reliable clinical test, because EHG recordings are long, noisy, subtly different between patients, and, crucially, scarce. Datasets of labeled recordings from women who later delivered preterm are small, which has made it difficult to train the data-hungry deep neural networks that have transformed other areas of medicine.

That scarcity and complexity, the authors argue, is precisely why most previous efforts relied on conventional machine learning pipelines, in which experts hand-craft features from the signal, such as entropy measures, frequency-band power ratios, or nonlinear descriptors, and then feed them to a classifier. Such approaches work, but they inherit the biases and blind spots of the features humans choose to extract. Deep learning, by contrast, can in principle discover discriminative patterns directly from raw data. The new study tackles the obstacles that have kept end-to-end deep learning out of the EHG field with a three-part strategy: a wavelet-based representation of the signal, an auxiliary feature that injects established physiological knowledge, and a training objective engineered to cope with severe class imbalance.

The first ingredient is the continuous wavelet transform, or CWT, a mathematical technique that decomposes a signal into a family of wavelets stretched and shifted across different scales. Unlike the Fourier transform, which tells you which frequencies are present in a recording but discards when they occurred, the wavelet transform preserves both time and frequency information simultaneously. Applied to an EHG recording, the CWT produces a two-dimensional image, a scalogram, in which the horizontal axis is time, the vertical axis is frequency, and brightness encodes the local energy of the signal. This transformation is a natural fit for uterine electrical activity, whose relevant patterns, such as the gradual downward shift of contraction-related frequencies, unfold over time. It also converts a one-dimensional signal into an image-like input, allowing the researchers to exploit the full power of convolutional neural networks, the same architecture that revolutionized image recognition.

On top of this time-frequency representation, the team added what they call an auxiliary feature: the peak amplitude, or PA, of the normalized power spectrum in the low-frequency band. This measure, highlighted in recent work as an effective standalone predictor of premature birth, captures how strongly the EHG energy is concentrated at the frequencies associated with preterm labor. By feeding this engineered feature into the network alongside the learned wavelet representations, CWT-AuxNet blends data-driven pattern discovery with domain knowledge accumulated over years of EHG research. The architecture itself is multibranch and convolutional, meaning parallel streams of filters process different aspects of the input before their outputs are merged, enabling the model to extract fine-grained features at the level of short signal windows rather than forcing a single judgment on an entire recording.

The third innovation addresses a problem that has quietly inflated results across the EHG literature: imbalance. Preterm deliveries are, fortunately, the minority outcome, so datasets contain far more term recordings than preterm ones. Naively trained classifiers tend to default to predicting the majority class, and oversampling techniques such as SMOTE, which synthesize artificial minority examples, have been shown in critical reanalyses to produce overly optimistic performance estimates. Instead of resampling the data, CWT-AuxNet uses a cost-sensitive loss function built on focal loss, a technique originally developed for dense object detection in computer vision. Focal loss down-weights the contribution of easy, well-classified examples and concentrates the gradient signal on hard, ambiguous ones, while class-specific weighting compensates for the rarity of preterm cases. The result is a network that learns to care about the minority class without fabricating synthetic data.

Because the model produces predictions for individual windows of the EHG recording rather than a single verdict per patient, the researchers designed a two-tier decision strategy. At the window level, the network assigns a risk score to each short segment of signal, capturing fine-grained fluctuations in uterine electrical behavior. At inference time, these window-level outputs are aggregated into a user-level decision through a dedicated strategy, yielding one individualized assessment of preterm risk per patient. This hierarchical design mirrors how a clinician might reason: noticing suspicious moments in a long monitoring session and then weighing them together to form an overall judgment. It also makes the system more robust, since a single noisy segment cannot dominate the final decision.

The performance numbers tell a compelling story. CWT-AuxNet achieved an AUC of 0.741 at the window level, indicating strong discrimination even when judging brief, isolated segments of signal where information is inherently limited. When window-level predictions were aggregated to the user level, performance climbed to an AUC of 0.932, meaning the model ranked individual patients’ preterm risk with high reliability. In head-to-head comparisons, the model consistently outperformed both traditional machine learning baselines built on hand-crafted features and earlier deep learning approaches. The authors also employed gradient-based visualization techniques, in the spirit of Grad-CAM, to probe which regions of the time-frequency representations drove the network’s decisions, offering a degree of interpretability that is essential for any technology hoping to enter prenatal care.

The implications reach beyond a single benchmark. Preterm birth prediction has long been dominated by clinical measures with limited predictive power when applied early: cervical length measured by ultrasound and fetal fibronectin testing, for example, show modest predictive value in threatened preterm labor. An EHG-based approach offers something different, a continuous, noninvasive window into the physiological maturation of the uterus itself, potentially usable in routine prenatal visits with standard surface electrodes. The study was supported by the National Natural Science Foundation of China, and the research team, led by co-first authors Xinliang Wen and Shengnan Zhuan with corresponding authors Lai Jiang and Xu Zhang, spans the University of Science and Technology of China, Bengbu Medical University, and the First Affiliated Hospital of USTC, combining expertise in microelectronics, life sciences, and obstetrics.

Challenges remain before CWT-AuxNet or any successor reaches the delivery ward. EHG datasets are still small and drawn largely from a limited number of recording centers, and the field has been burned before by methods that excelled on a single benchmark but failed to generalize. External validation on independent, multi-center cohorts, prospective clinical studies, and careful attention to calibration of risk scores will all be necessary. Yet the study marks a meaningful shift in how the problem is framed: rather than asking humans to define what distinguishes a preterm EHG recording, the wavelet-driven network learns those distinctions itself, guided by physiological priors and trained with an objective that respects the reality of imbalanced clinical data. If that approach holds up in the clinic, the faint electrical whispers of the uterus could become one of obstetrics’ most valuable early warning systems, giving mothers and doctors the most precious resource of all, time.

Subject of Research: Deep learning prediction of preterm birth from electrohysterogram signals

Article Title: A deep learning method for preterm birth prediction using wavelet representations and auxiliary features from electrohysterogram

Article References: Wen, X., Zhuan, S., Gao, X., Jiang, L., & Zhang, X. (2026). A deep learning method for preterm birth prediction using wavelet representations and auxiliary features from electrohysterogram. Medical & Biological Engineering & Computing. https://doi.org/10.1007/s11517-026-03602-3

Image Credits: AI Generated

DOI: 10.1007/s11517-026-03602-3

Keywords: preterm birth, electrohysterogram, deep learning, continuous wavelet transform, convolutional neural network, focal loss, class imbalance, uterine electromyography, predictive medicine, signal processing, noninvasive monitoring, machine learning