RAUIE: A Relation-Augmented Document-Level Event Extraction Model Based on UIE

In document-level event extraction tasks, multiple event types often coexist within a document (the multi-event problem), and arguments may play different roles across different events (the argument intertwining problem). Existing methods often suffer from significantly reduced event extraction accuracy, mainly when focusing on scenarios with scarce annotated data and imbalanced event-type distributions. This paper emphasizes enhancing the correlation between event triggers and arguments to address these challenges. Therefore, we propose the Relation-Augmented UIE Model (RADIE), which introduces entity relationships into prompts using prompt-based learning and designs a relation-augmentation algorithm to improve the accuracy and completeness of event extraction. Experimental results demonstrate that this approach significantly outperforms baseline methods, mainly showcasing excellent performance on datasets with long-tail distributions.