A possible solution is to apply beam search to enlarge the searching space at the first stage.
in our preliminary experiments, when the beam size is small, the diversity of predicted key facts is low, and also does not help to improve the accuracy.