Yuu Jinnai

Lipschitz Lifelong Reinforcement Learning

2021 · Proceedings of the AAAI Conference on Artificial Intelligence

We consider the problem of knowledge transfer when an agent is facing a series of Reinforcement Learning (RL) tasks. We introduce a novel metric between Markov Decision Processes and establish that close MDPs have close …

Model-Based Minimum Bayes Risk Decoding for Text Generation

2023 · arXiv (Cornell University)

Minimum Bayes Risk (MBR) decoding has been shown to be a powerful alternative to beam search decoding in a variety of text generation tasks. MBR decoding selects a hypothesis from a pool of hypotheses that …

Hyperparameter-Free Approach for Faster Minimum Bayes Risk Decoding

2024 · arXiv (Cornell University)

Minimum Bayes-Risk (MBR) decoding is shown to be a powerful alternative to beam search decoding for a wide range of text generation tasks. However, MBR requires a huge amount of time for inference to compute …