← Search

Yichao Du

16 accepted papers

2026

UMEM: Unified Memory Extraction and Management Framework for Generalizable Memory

ICML 2026poster

Self-evolving memory serves as the trainable parameters for Large Language Models (LLMs)-based agents, where extraction (distilling insights from experience) and management (updating the memory bank) must be tightly coordinated. Existing methods predominately optimize memory management while treatin…

Cited by 0SourceScholar
2026

UniRRM: Unified Reasoning Reward Models Across Languages and Evaluation Paradigms

ICML 2026poster

Reinforcement learning (RL) excels on tasks with verifiable rewards, but in open-ended tasks, the reliability of reward models remains a key challenge. Existing solutions either depend on costly proprietary LLM-as-a-Judge systems or opaque scalar reward models that lack interpretability. Recent work…

Cited by 0SourceScholar
2025

MJ-Bench: Is Your Multimodal Reward Model Really a Good Judge for Text-to-Image Generation?

NeurIPS 2025poster

While text-to-image models like GPT-4o-Image and FLUX are rapidly proliferating, they often encounter challenges such as hallucination, bias, and the production of unsafe, low-quality output. To effectively address these issues, it is crucial to align these models with desired behaviors based on fee…

Cited by 0SourcecodeScholar
2024

Communication-Efficient Personalized Federated Learning for Speech-to-Text Tasks

ICASSP 2024accepted

To protect privacy and meet legal regulations, federated learning (FL) has gained significant attention for training speech-to-text (S2T) systems, including automatic speech recognition (ASR) and speech translation (ST). However, the commonly used FL approach (i.e., FEDAVG) in S2T tasks typically su…

Cited by 0SourceScholar
2024

Federated Self-Explaining GNNs with Anti-shortcut Augmentations

ICML 2024poster

Graph Neural Networks (GNNs) have demonstrated remarkable performance in graph classification tasks. However, ensuring the explainability of their predictions remains a challenge. To address this, graph rationalization methods have been introduced to generate concise subsets of the original graph, k…

Cited by 0SourcePDFScholar
2024

Lost in the Source Language: How Large Language Models Evaluate the Quality of Machine Translation

ACL 2024findings

This study investigates how Large Language Models (LLMs) leverage source and reference data in machine translation evaluation task, aiming to better understand the mechanisms behind their remarkable performance in this task.We design the controlled experiments across various input modes and model ty…

2024

Towards Faithful Explanations: Boosting Rationalization with Shortcuts Discovery

ICLR 2024poster

The remarkable success in neural networks provokes the selective rationalization. It explains the prediction results by identifying a small subset of the inputs sufficient to support them. Since existing methods still suffer from adopting the shortcuts in data to compose rationales and limited large…

2023

Federated Nearest Neighbor Machine Translation

ICLR 2023poster

To protect user privacy and meet legal regulations, federated learning (FL) is attracting significant attention. Training neural machine translation (NMT) models with traditional FL algorithm (e.g., FedAvg) typically relies on multi-round model-based interactions. However, it is impractical and inef…

Cited by 12SourcePDFScholar
2023

IMTLab: An Open-Source Platform for Building, Evaluating, and Diagnosing Interactive Machine Translation Systems

EMNLP 2023long main

We present IMTLab, an open-source end-to-end interactive machine translation (IMT) system platform that enables researchers to quickly build IMT systems with state-of-the-art models, perform an end-to-end evaluation, and diagnose the weakness of systems. IMTLab treats the whole interactive translati…

Cited by 0SourcecodeScholar
2023

Nearest Neighbor Machine Translation is Meta-Optimizer on Output Projection Layer

EMNLP 2023long main

Nearest Neighbor Machine Translation ($k$NN-MT) has achieved great success in domain adaptation tasks by integrating pre-trained Neural Machine Translation (NMT) models with domain-specific token-level retrieval. However, the reasons underlying its success have not been thoroughly investigated. In…

Cited by 0SourcecodeScholar
2023

Simple and Scalable Nearest Neighbor Machine Translation

ICLR 2023poster

$k$NN-MT is a straightforward yet powerful approach for fast domain adaptation, which directly plugs the pre-trained neural machine translation (NMT) models with domain-specific token-level $k$-nearest-neighbor ($k$NN) retrieval to achieve domain adaptation without retraining. Despite being conceptu…

2022

DARE: Disentanglement-Augmented Rationale Extraction

NeurIPS 2022accept

Rationale extraction can be considered as a straightforward method of improving the model explainability, where rationales are a subsequence of the original inputs, and can be extracted to support the prediction results. Existing methods are mainly cascaded with the selector which extracts the ratio…

2022

Non-Parametric Domain Adaptation for End-to-End Speech Translation

EMNLP 2022main

The end-to-end speech translation (E2E-ST) has received increasing attention due to the potential of its less error propagation, lower latency and fewer parameters. However, the effectiveness of neural-based approaches to this task is severely limited by the available training corpus, especially for…

2022

Regularizing End-to-End Speech Translation with Triangular Decomposition Agreement

AAAI 2022technical

End-to-end speech-to-text translation (E2E-ST) is becoming increasingly popular due to the potential of its less error propagation, lower latency, and fewer parameters. Given the triplet training corpus〈speech, transcription, translation〉, the conventional high-quality E2E-ST system leverages the〈sp…

2021

Rethinking Zero-shot Neural Machine Translation: From a Perspective of Latent Variables

EMNLP 2021finding

Zero-shot translation, directly translating between language pairs unseen in training, is a promising capability of multilingual neural machine translation (NMT). However, it usually suffers from capturing spurious correlations between the output language and language invariant semantics due to the…