Deep learning models such as Bidirectional Encoder Representations from Transformers (BERT) achieve strong performance in text classification but struggle with uncertainty, especially in ambiguous or overlapping medical categories. We present BERT-TCPE, a lightweight framework that augments BERT with Term-Class Probability Embeddings (TCPE) derived from smoothed term–class likelihoods. Unlike ensemble-based methods that add computational cost, BERT-TCPE integrates probabilistic term semantics with contextual embeddings in a simple and efficient way. Experiments on two benchmark datasets—PubMed and Ohsumed—show consistent accuracy improvements over BERT, particularly in uncertain cases, without increasing inference time or parameter size. Embedding visualizations further indicate that BERT-TCPE enhances class separation. These results demonstrate that probabilistic-semantic augmentation provides an interpretable and scalable solution for improving robustness and decision confidence in medical Natural Language Processing.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

BERT-TCPE: A Term-Class Probability-Enhanced BERT Framework for Uncertainty-Aware Medical Text Classification

  • Prabhashrini Manage,
  • Yutong Wu,
  • Jinglan Zhang,
  • Yuefeng Li

摘要

Deep learning models such as Bidirectional Encoder Representations from Transformers (BERT) achieve strong performance in text classification but struggle with uncertainty, especially in ambiguous or overlapping medical categories. We present BERT-TCPE, a lightweight framework that augments BERT with Term-Class Probability Embeddings (TCPE) derived from smoothed term–class likelihoods. Unlike ensemble-based methods that add computational cost, BERT-TCPE integrates probabilistic term semantics with contextual embeddings in a simple and efficient way. Experiments on two benchmark datasets—PubMed and Ohsumed—show consistent accuracy improvements over BERT, particularly in uncertain cases, without increasing inference time or parameter size. Embedding visualizations further indicate that BERT-TCPE enhances class separation. These results demonstrate that probabilistic-semantic augmentation provides an interpretable and scalable solution for improving robustness and decision confidence in medical Natural Language Processing.