|
--- |
|
license: mit |
|
datasets: |
|
- McGill-NLP/FaithDial |
|
widget: |
|
- text: "A cardigan is a type of knitted garment (sweater) that has an open front. </s></s> The old version is the regular one, knitted garment that has open front and buttons!" |
|
model-index: |
|
- name: roberta-large-faithcritic |
|
results: |
|
- task: |
|
type: text-classification |
|
name: Faithfulness Critic |
|
dataset: |
|
name: 'FaithCritic' |
|
type: McGill-NLP/FaithDial |
|
metrics: |
|
- name: Accuracy |
|
type: accuracy |
|
value: 86.51 |
|
- task: |
|
type: text-classification |
|
name: Faithfulness Critic |
|
dataset: |
|
name: 'MNLI' |
|
type: multi_nli |
|
metrics: |
|
- name: Accuracy |
|
type: accuracy |
|
value: 74.720 |
|
- task: |
|
type: text-classification |
|
name: Faithfulness Critic |
|
dataset: |
|
name: 'BEGIN' |
|
type: begin |
|
metrics: |
|
- name: Accuracy |
|
type: accuracy |
|
value: 71.56 |
|
--- |
|
|
|
## Overview |
|
|
|
**Model Description:** roberta-large-faithcritic is the [RoBERTa large model](https://huggingface.co/roberta-large) fine-tuned on FaithCritic, a derivative of the [FaithDial](https://huggingface.co/datasets/McGill-NLP/FaithDial) dataset. The objective is to predict whether an utterance is faithful or not, given the source knowledge. |
|
|
|
The hyperparameters are provided in [hparams.yml](https://huggingface.co/McGill-NLP/roberta-large-faithcritic/blob/main/hparams.yaml). To know more about how to train a critic model, visit [our repo](https://github.com/McGill-NLP/FaithDial). |
|
|
|
## Usage |
|
|
|
```python |
|
from transformers import AutoTokenizer, AutoModel |
|
|
|
tokenizer = AutoTokenizer.from_pretrained("McGill-NLP/roberta-large-faithcritic") |
|
model = AutoModel.from_pretrained("McGill-NLP/roberta-large-faithcritic") |
|
|
|
knowledge = "A cardigan is a type of knitted garment (sweater) that has an open front." |
|
response = "The old version is the regular one, knitted garment that has open front and buttons!" |
|
input = tokenizer(knowledge, response) |
|
output = model(**input) |
|
|
|
``` |
|
|
|
|
|
## Citation Information |
|
|
|
```bibtex |
|
@article{dziri2022faithdial, |
|
title={FaithDial: A Faithful Benchmark for Information-Seeking Dialogue}, |
|
author={Dziri, Nouha and Kamalloo, Ehsan and Milton, Sivan and Zaiane, Osmar and Yu, Mo and Ponti, Edoardo and Reddy, Siva}, |
|
journal={arXiv preprint, arXiv:2204.10757}, |
|
year={2022}, |
|
url={https://arxiv.org/abs/2204.10757} |
|
} |
|
``` |
|
|