METHOD AND APPARATUS FOR IMPROVING QUALITY OF ATTENTION-BASED SEQUENCE-TO-SEQUENCE MODEL

摘要

A method and apparatus for improving the quality of an attention-based sequence-to-sequence model. The method includes determining an output sequence corresponding to an input sequence based on an attention-based sequence-to-sequence model, selecting at least one target attention head from among a plurality of attention heads, detecting at least one error output token among output tokens constituting the output sequence based on the target attention head, and correcting the output sequence based on the error output token.