GenIE: Generative Information Extraction
Martin JosifoskiNicola De CaoMaxime PeyrardFabio PetroniRobert West
Introduces the first end-to-end autoregressive framework for closed information extraction, using bi-level constrained generation to scale structured, knowledge-base-grounded fact extraction to millions of entities.
Extracting structured, factual knowledge from unstructured text into formal entity-and-relation statements is critical for artificial intelligence applications such as automated reasoning, knowledge discovery, and enterprise data management. However, traditional closed information extraction systems rely on multi-step pipeline architectures that sequentially identify entities, link them to an external database, and classify relations. These pipelines suffer from severe error accumulation across stages, fail under severe data imbalances, and break down when scaled beyond small, artificial schemas. Existing alternatives cannot handle realistic knowledge bases containing millions of entities and hundreds of relations.
The article introduces and evaluates GenIE (Generative Information Extraction), the first end-to-end generative system designed to perform closed information extraction at scale. The primary objective is to demonstrate that an autoregressive sequence-to-sequence model can directly generate structured factual statements from raw text while strictly adhering to large, predefined knowledge base schemas.
To achieve this, the approach employs a pre-trained sequence-to-sequence transformer model coupled with a bi-level constrained beam search decoding strategy. Instead of treating extraction as an unmanageable multi-class classification problem, GenIE dynamically generates entity and relation names in textual form. It enforces valid outputs using compact prefix search structures that restrict generation to valid schema components. The authors evaluated the system on large-scale datasets, including an annotated benchmark derived from Wikipedia and Wikidata containing nearly six million entities and 857 relations, alongside standard benchmarks such as Wiki-NRE, Geo-NRE, and FewRel.
The findings show that GenIE establishes a new state of the art, significantly outperforming prior end-to-end and multi-stage pipeline methods. On the large-scale benchmark, GenIE achieved an overall micro-F1 accuracy score of 68.9%, compared to 42.5% for a strong multi-step pipeline baseline—representing an approximate 60% relative improvement. Across underrepresented relations, GenIE demonstrated exceptional few-shot learning capability, reaching strong accuracy on relations with as few as 64 training examples, whereas the pipeline baseline required over 16,000 examples to achieve comparable quality. Furthermore, error disentanglement revealed that GenIE's unified architecture eliminates the severe initial drop-off caused by pipeline entity recognition errors, allowing entity and relation extraction to reinforce each other.
These results demonstrate that closed information extraction can transition from narrow academic setups to large-scale, real-world knowledge graphs. By replacing brittle multi-model pipelines with a single end-to-end model, organizations can reduce engineering complexity, maintenance overhead, and compounding pipeline errors. The ability to generalize from fewer data points per relation also lowers the substantial data annotation costs traditionally needed to expand schemas.
For practical adoption, organizations should consider deploying generative architectures for knowledge base population and semantic graph construction. Future technical work should focus on extending the constrained decoding mechanism to support literal text values—such as dates, measurements, and numerical properties—and expanding the system into multilingual extraction environments. While confidence in the experimental findings is high across multiple benchmarks, practitioners should note that performance still moderates on extremely rare relations, indicating that modest human-in-the-loop validation remains prudent for highly sensitive, low-frequency data domains.
- Paper: Generative Knowledge Graph Construction: A Review, Hongbin Ye et al. (2022). This review maps generative knowledge-graph construction approaches and benchmarks, clarifying the sequence-to-sequence design space in which GenIE is positioned.
- Paper: Unified Structure Generation for Universal Information Extraction, Yaojie Lu et al. (2022). UIE establishes a unified text-to-structure generation framework for information extraction, providing a direct conceptual foundation for GenIE’s generative extraction setup.
- Paper: Exploiting Asymmetry for Synthetic Training Data Generation: SynthIE and the Case of Information Extraction, Martin Josifoski et al. (2023). SynthIE follows GenIE by using synthetic, balanced training data to address its data-imbalance challenge and improve closed information extraction.
- Paper: Grammar-Constrained Decoding for Structured NLP Tasks without Finetuning, Saibo Geng et al. (2023). This work extends constrained generation for closed information extraction by applying input-dependent grammars to enforce valid outputs without task-specific fine-tuning.
- Paper: Extract, Define, Canonicalize: An LLM-based Framework for Knowledge Graph Construction, Bowen Zhang et al. (2024). EDC broadens schema-based knowledge-graph construction beyond GenIE’s fixed-schema extraction with free extraction, schema definition, and canonicalization.
