|
--- |
|
license: apache-2.0 |
|
language: en |
|
library: transformers |
|
other: distilbert |
|
datasets: |
|
- Short Question Answer Assessment Dataset |
|
--- |
|
|
|
# DistilBERT base uncased model for Short Question Answer Assessment |
|
|
|
## Model description |
|
|
|
DistilBERT is a transformers model, smaller and faster than BERT, which was pretrained on the same corpus in a |
|
self-supervised fashion, using the BERT base model as a teacher. This means it was pretrained on the raw texts only, |
|
with no humans labelling them in any way (which is why it can use lots of publicly available data) with an automatic |
|
process to generate inputs and labels from those texts using the BERT base model. |
|
|
|
This is a classification model that solves Short Question Answer Assessment task, finetuned [pretrained DistilBERT model](https://huggingface.co/distilbert-base-uncased) on |
|
[Fake and real news dataset](#) |
|
|
|
## Intended uses & limitations |
|
|
|
This can only be used for the kind of questions and answers provided by that are similar to the ones in the dataset of [Banjade et al.](https://aclanthology.org/W16-0520.pdf). |
|
|
|
|
|
### How to use |
|
|
|
You can use this model directly with a : |
|
|
|
```python |
|
>>> from transformers import pipeline |
|
>>> classifier = pipeline("text-classification", model="Giyaseddin/distilbert-base-uncased-finetuned-short-answer-assessment", return_all_scores=True) |
|
>>> context = "To rescue a child who has fallen down a well, rescue workers fasten him to a rope, the other end of which is then reeled in by a machine. The rope pulls the child straight upward at steady speed." |
|
>>> question = "How does the amount of tension in the rope compare to the downward force of gravity acting on the child?" |
|
>>> ref_answer = "Since the child is being raised straight upward at a constant speed, the net force on the child is zero and all the forces balance. That means that the tension in the rope balances the downward force of gravity." |
|
>>> student_answer = "The tension force is higher than the force of gravity." |
|
>>> |
|
>>> body = " [SEP] ".join([context, question, ref_answer, student_answer]) |
|
>>> raw_results = classifier([body]) |
|
>>> raw_results |
|
[[{'label': 'LABEL_0', 'score': 0.0004029414849355817}, |
|
{'label': 'LABEL_1', 'score': 0.0005476847873069346}, |
|
{'label': 'LABEL_2', 'score': 0.998059093952179}, |
|
{'label': 'LABEL_3', 'score': 0.0009902542224153876}]] |
|
>>> _LABELS_ID2NAME = {0: "correct", 1: "correct_but_incomplete", 2: "contradictory", 3: "incorrect"} |
|
>>> results = [] |
|
>>> for result in raw_results: |
|
for score in result: |
|
results.append([ |
|
{_LABELS_ID2NAME[int(score["label"][-1:])]: "%.2f" % score["score"]} |
|
]) |
|
|
|
>>> results |
|
[[{'correct': '0.00'}], |
|
[{'correct_but_incomplete': '0.00'}], |
|
[{'contradictory': '1.00'}], |
|
[{'incorrect': '0.00'}]] |
|
``` |
|
|
|
### Limitations and bias |
|
|
|
Even if the training data used for this model could be characterized as fairly neutral, this model can have biased |
|
predictions. It also inherits some of |
|
[the bias of its teacher model](https://huggingface.co/bert-base-uncased#limitations-and-bias). |
|
|
|
This bias will also affect all fine-tuned versions of this model. |
|
|
|
Also one of the limiations of this model is the length, longer sequences would lead to wrong predictions, due to the pre-processing phase (after concatentating the input sequences, the important student answer might be pruned!) |
|
|
|
## Pre-training data |
|
|
|
DistilBERT pretrained on the same data as BERT, which is [BookCorpus](https://yknzhu.wixsite.com/mbweb), a dataset |
|
consisting of 11,038 unpublished books and [English Wikipedia](https://en.wikipedia.org/wiki/English_Wikipedia) |
|
(excluding lists, tables and headers). |
|
|
|
## Fine-tuning data |
|
|
|
The annotated dataset consists of 900 students’ short constructed answers and their correctness in the given context. Four qualitative levels of correctness are defined, correct, correct-but-incomplete, contradictory and Incorrect. |
|
|
|
|
|
## Training procedure |
|
|
|
### Preprocessing |
|
|
|
In the preprocessing phase, the following parts are concatenated: _question context_, _question_, _reference_answer_, and _student_answer_ using the separator `[SEP]`. |
|
This makes the full text as: |
|
|
|
``` |
|
[CLS] Context Sentence [SEP] Question Sentence [SEP] Reference Answer Sentence [SEP] Student Answer Sentence [CLS] |
|
``` |
|
|
|
The data are splitted according to the following ratio: |
|
- Training set 80%. |
|
- Test set 20%. |
|
|
|
Lables are mapped as: `{0: "correct", 1: "correct_but_incomplete", 2: "contradictory", 3: "incorrect"}` |
|
|
|
### Fine-tuning |
|
|
|
The model was finetuned on GeForce GTX 960M for 20 minuts. The parameters are: |
|
|
|
| Parameter | Value | |
|
|:-------------------:|:-----:| |
|
| Learning rate | 5e-5 | |
|
| Weight decay | 0.01 | |
|
| Training batch size | 8 | |
|
| Epochs | 4 | |
|
|
|
Here is the scores during the training: |
|
|
|
|
|
| Epoch | Training Loss | Validation Loss | Accuracy | F1 | Precision | Recall | |
|
|:----------:|:-------------:|:-----------------:|:----------:|:---------:|:----------:|:--------:| |
|
| 1 | No log | 0.665765 | 0.755330 | 0.743574 | 0.781210 | 0.755330 | |
|
| 2 | 0.932100 | 0.362124 | 0.890355 | 0.889875 | 0.891407 | 0.890355 | |
|
| 3 | 0.364900 | 0.226225 | 0.942132 | 0.941802 | 0.942458 | 0.942132 | |
|
| 3 | 0.176900 | 0.193660 | 0.954315 | 0.954175 | 0.954985 | 0.954315 | |
|
|
|
## Evaluation results |
|
|
|
When fine-tuned on downstream task of fake news binary classification, this model achieved the following results: |
|
(scores are rounded to 2 floating points) |
|
|
|
|
|
| | precision | recall | f1-score | support | |
|
|:------------------------:|:----------:|:-------:|:--------:|:-------:| |
|
| _correct_ | 0.938 | 0.989 | 0.963 | 366 | |
|
| _correct_but_incomplete_ | 0.975 | 0.922 | 0.948 | 257 | |
|
| _contradictory_ | 0.946 | 0.938 | 0.942 | 113 | |
|
| _incorrect_ | 0.963 | 0.944 | 0.953 | 249 | |
|
| accuracy | - | - | 0.954 | 985 | |
|
| macro avg | 0.956 | 0.948 | 0.952 | 985 | |
|
| weighted avg | 0.955 | 0.954 | 0.954 | 985 | |
|
|
|
Confision matrix: |
|
|
|
|
|
|
|
| Actual \ Predicted | _correct_ | _correct_but_incomplete_ | _contradictory_ | _incorrect_ | |
|
|:------------------------:|:---------:|:------------------------:|:---------------:|:-----------:| |
|
| _correct_ | 362 | 4 | 0 | 0 | |
|
| _correct_but_incomplete_ | 13 | 237 | 0 | 7 | |
|
| _contradictory_ | 4 | 1 | 106 | 2 | |
|
| _incorrect_ | 7 | 1 | 6 | 235 | |
|
|
|
|
|
|
|
The AUC score is: 'micro'= **0.9695** and 'macro': **0.9659** |
|
|
