HyperPIE: Hyperparameter Information Extraction from Scientific Publications

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Saier, Tarek, Ohta, Mayumi, Asakura, Takuto, Färber, Michael
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910393630195712
author Saier, Tarek
Ohta, Mayumi
Asakura, Takuto
Färber, Michael
author_facet Saier, Tarek
Ohta, Mayumi
Asakura, Takuto
Färber, Michael
contents Automatic extraction of information from publications is key to making scientific knowledge machine readable at a large scale. The extracted information can, for example, facilitate academic search, decision making, and knowledge graph construction. An important type of information not covered by existing approaches is hyperparameters. In this paper, we formalize and tackle hyperparameter information extraction (HyperPIE) as an entity recognition and relation extraction task. We create a labeled data set covering publications from a variety of computer science disciplines. Using this data set, we train and evaluate BERT-based fine-tuned models as well as five large language models: GPT-3.5, GALACTICA, Falcon, Vicuna, and WizardLM. For fine-tuned models, we develop a relation extraction approach that achieves an improvement of 29% F1 over a state-of-the-art baseline. For large language models, we develop an approach leveraging YAML output for structured data extraction, which achieves an average improvement of 5.5% F1 in entity recognition over using JSON. With our best performing model we extract hyperparameter information from a large number of unannotated papers, and analyze patterns across disciplines. All our data and source code is publicly available at https://github.com/IllDepence/hyperpie
format Preprint
id arxiv_https___arxiv_org_abs_2312_10638
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle HyperPIE: Hyperparameter Information Extraction from Scientific Publications
Saier, Tarek
Ohta, Mayumi
Asakura, Takuto
Färber, Michael
Computation and Language
Information Retrieval
Automatic extraction of information from publications is key to making scientific knowledge machine readable at a large scale. The extracted information can, for example, facilitate academic search, decision making, and knowledge graph construction. An important type of information not covered by existing approaches is hyperparameters. In this paper, we formalize and tackle hyperparameter information extraction (HyperPIE) as an entity recognition and relation extraction task. We create a labeled data set covering publications from a variety of computer science disciplines. Using this data set, we train and evaluate BERT-based fine-tuned models as well as five large language models: GPT-3.5, GALACTICA, Falcon, Vicuna, and WizardLM. For fine-tuned models, we develop a relation extraction approach that achieves an improvement of 29% F1 over a state-of-the-art baseline. For large language models, we develop an approach leveraging YAML output for structured data extraction, which achieves an average improvement of 5.5% F1 in entity recognition over using JSON. With our best performing model we extract hyperparameter information from a large number of unannotated papers, and analyze patterns across disciplines. All our data and source code is publicly available at https://github.com/IllDepence/hyperpie
title HyperPIE: Hyperparameter Information Extraction from Scientific Publications
topic Computation and Language
Information Retrieval
url https://arxiv.org/abs/2312.10638