Message Passing Attention Networks for Document Understanding
Table 2: Classification accuracies. Best performance per column in bold, *best MPAD variant. OOM: >16GB RAM.
['[BOLD] Model', 'Reut.', 'BBC', 'Pol.', 'Subj.', 'MPQA', 'IMDB', 'TREC', 'SST-1', 'SST-2', 'Yelp’13']
: Ablation results. The n in nMP refers to the number of message passing iterations. For the baselines, we provide the scores reported in the original papers. Furthermore, we have evaluated some of the baselines on the rest of our benchmark datasets, and we also report these scores. MPAD reaches best performance on 5 out of 10 datasets, and is close second elsewhere. Moreover, the 5 datasets on which MPAD ranks first widely differ in training set size, number of categories, and prediction task (topic, sentiment, etc.), which indicates that MPAD can perform well in different settings.