TocBERT: Medical Document Structure Extraction Using Bidirectional Transformers

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Saleh, Majd, Baghdadi, Sarra, Paquelet, Stéphane
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909300132151296
author Saleh, Majd
Baghdadi, Sarra
Paquelet, Stéphane
author_facet Saleh, Majd
Baghdadi, Sarra
Paquelet, Stéphane
contents Text segmentation holds paramount importance in the field of Natural Language Processing (NLP). It plays an important role in several NLP downstream tasks like information retrieval and document summarization. In this work, we propose a new solution, namely TocBERT, for segmenting texts using bidirectional transformers. TocBERT represents a supervised solution trained on the detection of titles and sub-titles from their semantic representations. This task was formulated as a named entity recognition (NER) problem. The solution has been applied on a medical text segmentation use-case where the Bio-ClinicalBERT model is fine-tuned to segment discharge summaries of the MIMIC-III dataset. The performance of TocBERT has been evaluated on a human-labeled ground truth corpus of 250 notes. It achieved an F1-score of 84.6% when evaluated on a linear text segmentation problem and 72.8% on a hierarchical text segmentation problem. It outperformed a carefully designed rule-based solution, particularly in distinguishing titles from subtitles.
format Preprint
id arxiv_https___arxiv_org_abs_2406_19526
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TocBERT: Medical Document Structure Extraction Using Bidirectional Transformers
Saleh, Majd
Baghdadi, Sarra
Paquelet, Stéphane
Computation and Language
Information Retrieval
Machine Learning
Text segmentation holds paramount importance in the field of Natural Language Processing (NLP). It plays an important role in several NLP downstream tasks like information retrieval and document summarization. In this work, we propose a new solution, namely TocBERT, for segmenting texts using bidirectional transformers. TocBERT represents a supervised solution trained on the detection of titles and sub-titles from their semantic representations. This task was formulated as a named entity recognition (NER) problem. The solution has been applied on a medical text segmentation use-case where the Bio-ClinicalBERT model is fine-tuned to segment discharge summaries of the MIMIC-III dataset. The performance of TocBERT has been evaluated on a human-labeled ground truth corpus of 250 notes. It achieved an F1-score of 84.6% when evaluated on a linear text segmentation problem and 72.8% on a hierarchical text segmentation problem. It outperformed a carefully designed rule-based solution, particularly in distinguishing titles from subtitles.
title TocBERT: Medical Document Structure Extraction Using Bidirectional Transformers
topic Computation and Language
Information Retrieval
Machine Learning
url https://arxiv.org/abs/2406.19526