Our approach can, in principle, be applied to any classification task that is well modeled by jointly solving an extraction subtask.
as evidenced by our experiments, proper training does require a reasonable initial guess of the extracted ex-planations, as well as ways to mitigate the risk of the extraction subtask suppressing too much information (such as via feature smoothing).