Skip to content

[CoNLL23] JaSPICE: Automatic Evaluation Metric Using Predicate-Argument Structures for Image Captioning Models

License

Notifications You must be signed in to change notification settings

keio-smilab23/JaSPICE

Repository files navigation

[CoNLL23] JaSPICE: Automatic Evaluation Metric Using Predicate-Argument Structures for Image Captioning Models

Lint Test

Image captioning studies heavily rely on automatic evaluation metrics such as BLEU and METEOR. However, such n-gram-based metrics have been shown to correlate poorly with human evaluation, leading to the proposal of alternative metrics such as SPICE for English; however, no equivalent metrics have been established for other languages. Therefore, in this study, we propose an automatic evaluation metric called JaSPICE, which evaluates Japanese captions based on scene graphs. The proposed method generates a scene graph from dependencies and the predicate-argument structure, and extends the graph using synonyms. We conducted experiments employing 10 image captioning models trained on STAIR Captions and PFN-PIC and constructed the Shichimi dataset, which contains 103,170 human evaluations. The results showed that our metric outperformed the baseline metrics for the correlation coefficient with the human evaluation.

system

Instructions (using Docker)

Clone & Install

pip install jaspice

or

git clone git@github.com:keio-smilab23/JaSPICE.git
cd JaSPICE
pip install -e .

Build

docker build -t jaspice .

Run the docker container.

docker run -d -p 2115:2115 jaspice

Usage

from jaspice.api import JaSPICE

batch_size = 16
jaspice = JaSPICE(batch_size,server_mode=True)
_, score = jaspice.compute_score(references, candidates)

Instructions (without Docker)

Clone & Install

git clone git@github.com:keio-smilab23/JaSPICE.git
cd JaSPICE
pip install -e .

Install JUMAN, JUMAN++, KNP

  • juman : v7.01
  • juman++ : v1.02
  • knp : v4.20
# JUMAN++
wget 'https://github.com/keio-smilab23/JaSPICE/releases/download/0.0.1/jumanpp.tar.xz'

# JUMAN
wget 'https://github.com/keio-smilab23/JaSPICE/releases/download/0.0.1/juman.tar.bz2'

# KNP
wget 'https://github.com/keio-smilab23/JaSPICE/releases/download/0.0.1/knp.tar.bz2'

Usage

from jaspice.api import JaSPICE

batch_size = 16
jaspice = JaSPICE(batch_size,server_mode=False)
_, score = jaspice.compute_score(references, candidates)

Scene Graph Example

  • 「人通りの少なくなった道路で青いズボンを着た男の子がオレンジ色のヘルメットを被りスケートボードに乗っている.」

BibTex

@inproceedings{wada2023,
   title     = {{JaSPICE: Automatic Evaluation Metric Using Predicate-Argument Structures for Image Captioning Models}},
   author    = {Wada, Yuiga  and Kaneda, Kanta  and Sugiura, Komei},
   year      = 2023,
   booktitle = {Proceedings of the 27th Conference on Computational Natural Language Learning (CoNLL)}
}

Others

Licenses

This work is licensed under the BSD-3-Clause-Clear license. To view a copy of this license, see LICENSE.

About

[CoNLL23] JaSPICE: Automatic Evaluation Metric Using Predicate-Argument Structures for Image Captioning Models

Resources

License

Stars

Watchers

Forks

Packages

No packages published