ETC: An Ensemble Transformer Framework for Fetal Health Classification

Fetal health monitoring and assessment play a crucial role in reducing perinatal mortality and ensuring safe childbirth outcomes. While traditional machine learning approaches to fetal health classification have shown promise, they often struggle with capturing complex temporal dependencies and subtle feature interactions in Cardiotocography (CTG) data. To address these limitations, we propose an Ensemble Transformer Classification (ETC) framework that leverages the power of multiple specialized Transformer encoders for enhanced fetal health assessment. Our framework incorporates three key innovations: (1) a dual encoding mechanism combining positional and semantic encoding to better represent CTG features, (2) a multi-head attention mechanism with dynamic weighting for adaptive feature interaction modeling, and (3) a Bagging-based ensemble strategy to improve model robustness and reduce prediction variance. Experimental results on CTG data demonstrate that our ETC framework achieves 94.37% overall accuracy, with particularly strong performance in identifying pathological cases (96% Precision, 93% Recall), significantly outperforming traditional machine learning methods and standard deep learning approaches. The framework shows substantial practical value for clinical diagnostic assistance, telemedicine applications, and medical education. Furthermore, its attention mechanism visualization provides interpretable insights into feature importance, making it particularly suitable for real-world medical applications.

Authors

Institutions

Publication Details

Published
2024-12-06
DOI
https://doi.org/10.1109/iaecst64597.2024.11117623
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

ETC: An Ensemble Transformer Framework for Fetal Health Classification

Tianhang Chen, Ka-Chun Wong, Zhongyu Yao
Machine Learning in Healthcare
article

ETC: An Ensemble Transformer Framework for Fetal Health Classification

Tianhang Chen, Ka-Chun Wong, Zhongyu Yao
article en

Abstract

Fetal health monitoring and assessment play a crucial role in reducing perinatal mortality and ensuring safe childbirth outcomes. While traditional machine learning approaches to fetal health classification have shown promise, they often struggle with capturing complex temporal dependencies and subtle feature interactions in Cardiotocography (CTG) data. To address these limitations, we propose an Ensemble Transformer Classification (ETC) framework that leverages the power of multiple specialized Transformer encoders for enhanced fetal health assessment. Our framework incorporates three key innovations: (1) a dual encoding mechanism combining positional and semantic encoding to better represent CTG features, (2) a multi-head attention mechanism with dynamic weighting for adaptive feature interaction modeling, and (3) a Bagging-based ensemble strategy to improve model robustness and reduce prediction variance. Experimental results on CTG data demonstrate that our ETC framework achieves 94.37% overall accuracy, with particularly strong performance in identifying pathological cases (96% Precision, 93% Recall), significantly outperforming traditional machine learning methods and standard deep learning approaches. The framework shows substantial practical value for clinical diagnostic assistance, telemedicine applications, and medical education. Furthermore, its attention mechanism visualization provides interpretable insights into feature importance, making it particularly suitable for real-world medical applications.

City University of Hong Kong (HK)
Gender equality
Openalex Percentile: Top 72%
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.