The basic factorized model got it wrong, assigning A1 to the argument 'state'.
this work is similar to energy network approach (Belanger and McCallum, 2016), while a global score function is provided, and approximate inference steps are used.