Multi-task learning for interpretable cause of death classification using key phrase prediction

Abstract

We introduce a multi-task learning model for cause-of-death classification of verbal autopsy narratives that jointly learns to output interpretable key phrases. Adding these key phrases outperforms the baseline model and topic modeling features.

Tasks

Reproductions