DiffusionSL: Sequence Labeling via Tag Diffusion Process

Ziyang Huang, Pengfei Cao, Jun Zhao, Kang Liu


Abstract
Sequence Labeling (SL) is long-standing in Natural Language Processing (NLP). Traditionally, discriminative models have been widely used to capture the conditional distribution of sequence tags, rather than generative models. In this paper, we present DiffusionSL, a framework that utilizes a conditional discrete diffusion model for generating discrete tag data, resulting in a Tag Diffusion Process. We treat the natural language sequence as the conditional signal and the sequence tags as the generation target, iteratively refining the noisy tags to obtain clean ones. To address the discreteness issue, we propose the Bit-Tag Converter (BTConverter) to model the target in continuous data space. Furthermore, we introduce the Bit Diffusion Transformer (BitDiT) to model the process of noise elimination. Leveraging the powerful iterative refinement capability of the diffusion model, DiffusionSL achieves superior performance against previous state-of-the-art (SOTA) baselines and outperforms gpt-3.5-turbo significantly across multiple benchmark datasets and various tasks.
Anthology ID:
2023.findings-emnlp.860
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2023
Month:
December
Year:
2023
Address:
Singapore
Editors:
Houda Bouamor, Juan Pino, Kalika Bali
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
12902–12920
Language:
URL:
https://aclanthology.org/2023.findings-emnlp.860
DOI:
10.18653/v1/2023.findings-emnlp.860
Bibkey:
Cite (ACL):
Ziyang Huang, Pengfei Cao, Jun Zhao, and Kang Liu. 2023. DiffusionSL: Sequence Labeling via Tag Diffusion Process. In Findings of the Association for Computational Linguistics: EMNLP 2023, pages 12902–12920, Singapore. Association for Computational Linguistics.
Cite (Informal):
DiffusionSL: Sequence Labeling via Tag Diffusion Process (Huang et al., Findings 2023)
Copy Citation:
PDF:
https://aclanthology.org/2023.findings-emnlp.860.pdf