[TIL] Experiments: Increasing accuracy of RNN model

06/16/23

·

3 min read

[TIL] Experiments: Increasing accuracy of RNN model

I conducted four experiments after creating a normalized dataset, and I found that the accuracy significantly improved. Additionally, I attempted cumulative learning. Here are the details of each experiment:

Experiment 1: Cumulative Learning ❌ with a dataset of around 900 instances

I generated a dataset of around 900 sentence examples based on selected symptoms with levels 1 to 5 using KTAS (Korean Triage and Acuity Scale: Emergency Level Standard) and performed tests.

Experiment 1 Results

Experiment 1 Results

Experiment 1 Results

Experiment 1 Results

The model accurately predicted the acuity levels only for the symptoms it was trained on (indicated by the blue boxes). However, it struggled to make accurate judgments for sentences containing symptoms that were not included in the training.

Experiment 2: Cumulative Learning 🔺 with the initial model (dataset of around 900 instances) + cumulative model (new dataset of around 800 instances)

I selected additional symptoms and generated around 800 sentence examples. I then performed cumulative learning on the model trained in Experiment 1.

Experiment 2 Results

Experiment 2 Results

Experiment 2 Results

Experiment 2 Results

The accuracy decreased compared to the model trained in the first experiment. Despite adding sentences related to symptoms in the additional training dataset, there were still errors in acuity prediction. Particularly, in cases indicated by the red boxes, where an acuity level of 5 was expected but the model predicted 1 with high confidence.

Experiment 3: Cumulative Learning ❌ with a dataset of around 1700 instances

Due to the decreased accuracy and insufficient severity judgment in the model trained in Experiment 2, I created a new dataset of around 1700 sentence examples and trained a new model.

Experiment 3 Results

Experiment 3 Results

Experiment 3 Results

Since this experiment was similar to Experiment 1, the model performed well in predicting acuity for all sentences that included the symptoms it was trained on.

Based on this finding, I conducted Experiment 4.

Experiment 4: Cumulative Learning ⭕ with the initial model (dataset of around 900 instances) + cumulative model (new dataset of around 1700 instances)

Considering the improvements made in Experiment 3, where a single model was trained with a dataset of around 1700 instances, I applied a similar approach to Experiment 4. I performed cumulative learning by combining the initial model trained in Experiment 1 (dataset of around 900 instances) with the cumulative model trained on the new dataset of around 1700 instances.

Experiment 4 Results

Experiment 4 Results

Experiment 4 Results

Experiment 4 Results

Compared to the accuracy of the initial model trained in Experiment 1 (accuracy: 0.927), the accuracy increased to 0.969. The acuity prediction also improved. However, there was a slight decrease in acuity prediction compared to Experiment 3. The confidence score also showed 91% accuracy in incorrectly predicted sentences.