Scalable and Domain-General Abstractive Proposition Segmentation

TL;DR

This paper introduces a scalable, supervised approach to abstractive proposition segmentation using fine-tuned large language models, improving accuracy and domain generalization for NLP applications.

Contribution

It presents a supervised training method for proposition segmentation with LLMs, leveraging teacher-student models and synthetic data for improved scalability and domain adaptation.

Findings

Supervised training with annotated datasets improves segmentation accuracy.

Using teacher models to generate synthetic data enables training smaller, effective models.

The approach demonstrates strong domain generalization capabilities.

Abstract

Segmenting text into fine-grained units of meaning is important to a wide range of NLP applications. The default approach of segmenting text into sentences is often insufficient, especially since sentences are usually complex enough to include multiple units of meaning that merit separate treatment in the downstream task. We focus on the task of abstractive proposition segmentation (APS): transforming text into simple, self-contained, well-formed sentences. Several recent works have demonstrated the utility of proposition segmentation with few-shot prompted LLMs for downstream tasks such as retrieval-augmented grounding and fact verification. However, this approach does not scale to large amounts of text and may not always extract all the facts from the input text. In this paper, we first introduce evaluation metrics for the task to measure several dimensions of quality. We then propose a scalable, yet accurate, proposition segmentation model. We model proposition segmentation as a supervised task by training LLMs on existing annotated datasets and show that training yields significantly improved results. We further show that by using the fine-tuned LLMs (Gemini Pro and Gemini Ultra) as teachers for annotating large amounts of multi-domain synthetic distillation data, we can train smaller student models (Gemma 1 2B and 7B) with results similar to the teacher LLMs. We then demonstrate that our technique leads to effective domain generalization, by annotating data in two domains outside the original training data and evaluating on them. Finally, as a key contribution of the paper, we share an easy-to-use API for NLP practitioners to use.