Active learning (AL) has demonstrated remarkable potential in reducing the annotation effort required for training machine learning models. However, despite the surging popularity of natural language generation (NLG) tasks in recent years, the application of AL to NLG has been limited. In this paper, we introduce Active Text Generation (ATGen) - a comprehensive framework that bridges AL with text generation tasks, enabling the application of state-of-the-art AL strategies to NLG. Our framework simplifies AL-empowered annotation in NLG tasks using both human annotators and automatic annotation agents based on large language models (LLMs). The framework supports LLMs deployed as services, such as ChatGPT and Claude, or operated on-premises. Furthermore, ATGen provides a unified platform for smooth implementation and benchmarking of novel AL strategies tailored to NLG tasks. Finally, we present evaluation results for state-of-the-art AL strategies across diverse settings and multiple text generation tasks. We show that ATGen reduces both the effort of human annotators and costs associated with API calls to LLM-based annotation agents. The code of the framework is available on GitHub under the MIT license. The video presentation is available at http://atgen-video.nlpresearch.group
@article{arxiv.2506.23342,
title = {ATGen: A Framework for Active Text Generation},
author = {Akim Tsvigun and Daniil Vasilev and Ivan Tsvigun and Ivan Lysenko and Talgat Bektleuov and Aleksandr Medvedev and Uliana Vinogradova and Nikita Severin and Mikhail Mozikov and Andrey Savchenko and Rostislav Grigorev and Ramil Kuleev and Fedor Zhdanov and Artem Shelmanov and Ilya Makarov},
journal= {arXiv preprint arXiv:2506.23342},
year = {2025}
}