Giyaseddin's picture
Update README.md
4904ea2
metadata
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 on Question Answer Assessment 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..

How to use

You can use this model directly with a :

>>> 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.

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, a dataset consisting of 11,038 unpublished books and 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 Question Answer Assessment, 4 class 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

Confusion 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