Paper: "LLMs Can Also Do Well! Breaking Barriers in Semantic Role Labeling via Large Language Models"
Published: ACL Findings 2025
Authors: Xinxin Li*, Huiyao Chen*, Chengjun Liu, Jing Li, Meishan Zhangβ , Jun Yu, Min Zhang
This project presents a novel two-stage framework for Semantic Role Labeling (SRL) using Large Language Models. By combining Retrieval-Augmented Agent and Self-Correction Mechanism, we achieve state-of-the-art performance on CPB1.0, CoNLL-2009, and CoNLL-2012 datasets.
- Retrieval-Augmented Agent - Injects external predicate-argument frame knowledge during predicate identification and argument labeling
- Self-Correction Mechanism - Trains LLMs to identify and iteratively correct inconsistencies in their own SRL outputs
- Conversation-Based Two-Stage Architecture - Stage 1: Predicate identification; Stage 2: Argument labeling
π TODO
- Underway in the process of changing the paths to relative paths.
Basic Environment:
pip install -r requirements.txtvLLM Acceleration Environment (Optional, for faster inference):
# Create a separate conda environment
conda create -n vllm-env python=3.8
conda activate vllm-env
pip install -r vllm_requirements.txt| Dataset | Language | Source |
|---|---|---|
| CPB1.0 | Chinese | LDC2005T23 |
| CoNLL-2009 | Chinese & English | LDC2012T03 |
| CoNLL-2012 | English | Official Site |
The processed data format is JSON with the following structure:
{
"text": "sentence text",
"srl": [
{
"pred": "predicate",
"position": [start_pos, end_pos],
"arguments": [
{
"value": "argument_value",
"position": [start_pos, end_pos],
"role": "role_label"
}
]
}
],
"token": ["tokenized", "words"],
"pos": ["part-of-speech", "tags"], // Required for CoNLL datasets
"lemmas": ["lemmatized", "forms"] // Required for CoNLL datasets
}Note: Position indices use 1-based numbering (starting from 1)
Run the appropriate scripts based on your dataset:
English Datasets (CoNLL-2012):
python agent_scripts/construct_database_conll12_en.py
python agent_scripts/construct_agent_conll12_en.pyChinese Datasets (CPB1.0, CoNLL-2009 Chinese):
python agent_scripts/construct_database_zh.py
python gpt_infer.py
python get_chinese_pred_des.py# Run the instruction generation script for your dataset
python agent_scripts/generate_instructions_[dataset].py- Edit Training Configuration:
vim examples/train_lora/llama3_lora_sft_ds0.yaml- Start Training:
bash train.shChoose your inference method:
Standard Inference:
python chat_[dataset].pyvLLM Accelerated Inference:
python chat_[dataset]_vllm.pyPost-processing:
# For English datasets
python process_rl_en.py
# For Chinese datasets
python process_rl_zh.pyEvaluation:
python metric.pyLLM-SRL/
βββ agent_scripts/ # Agent construction scripts
βββ examples/ # Training configuration examples
βββ assets/ # Project assets
βββ requirements.txt # Basic dependencies
βββ vllm_requirements.txt # vLLM dependencies
βββ chat_*.py # Inference scripts
βββ process_rl_*.py # Post-processing scripts
βββ metric.py # Evaluation script
- First-time Users: Recommend testing the complete pipeline on a small dataset first
- Resource Requirements: Training requires substantial GPU memory; inference can be accelerated with vLLM
- Dataset Selection: Choose appropriate language and dataset based on your task requirements
- Parameter Tuning: Adjust parameters in the training configuration file according to your specific task
Our method achieves state-of-the-art results on:
- CPB1.0 (Chinese Proposition Bank)
- CoNLL-2009 (Chinese & English)
- CoNLL-2012 (English)
The two-stage framework with retrieval augmentation and self-correction significantly outperforms previous generative LLM approaches for SRL tasks.
- Provides external predicate-argument frame knowledge
- Enhances semantic accuracy during both stages
- Reduces hallucinations in argument identification
- Iteratively refines LLM outputs
- Identifies and corrects inconsistencies
- Improves overall labeling consistency
- Stage 1: Predicate identification with context understanding
- Stage 2: Argument labeling with enhanced reasoning
- Both stages benefit from retrieval and self-correction
If this work is helpful to your research, please cite:
@article{DBLP:journals/corr/abs-2506-05385,
author = {Xinxin Li and Huiyao Chen and Chengjun Liu and
Jing Li and Meishan Zhang and Jun Yu and Min Zhang},
title = {LLMs Can Also Do Well! Breaking Barriers in Semantic Role Labeling
via Large Language Models},
year = {2025},
url = {https://doi.org/10.48550/arXiv.2506.05385},
}- π§ Contact: Please submit issues or contact the authors for questions
- π Star Support: If this project helps you, please give it a star
- π Pull Requests: Contributions and improvements are welcome
We thank the developers of the datasets and the open-source community for their valuable contributions to semantic role labeling research.
