← Search

Yujia Xie

15 accepted papers

2026

Improving Code Localization with Repository Memory

ICLR 2026poster

Code localization is a fundamental challenge in repository-level software engineering tasks such as bug fixing. While existing methods equip language agents with comprehensive tools/interfaces to fetch information from the repository, they overlook the critical aspect of *memory*, where each instanc…

Cited by 0SourceScholar
2025

Developing a Reliable, Fast, General-Purpose Hallucination Detection and Mitigation Service

NAACL 2025industry

Hallucination, a phenomenon where large language models (LLMs) produce output that is factually incorrect or unrelated to the input, is a major challenge for LLM applications that require accuracy and dependability. In this paper, we introduce a reliable and high-speed production system aimed at det…

Cited by 0SourcePDFScholar
2024

DoLa: Decoding by Contrasting Layers Improves Factuality in Large Language Models

ICLR 2024poster

Despite their impressive capabilities, large language models (LLMs) are prone to hallucinations, i.e., generating content that deviates from facts seen during pretraining. We propose a simple decoding strategy for reducing hallucinations with pretrained LLMs that does not require conditioning on ret…

2023

Improving Commonsense in Vision-Language Models via Knowledge Graph Riddles

CVPR 2023highlight

This paper focuses on analyzing and improving the commonsense ability of recent popular vision-language (VL) models. Despite the great success, we observe that existing VL-models still lack commonsense knowledge/reasoning ability (e.g., "Lemons are sour"), which is a vital component towards artifici…

2023

Look Before You Match: Instance Understanding Matters in Video Object Segmentation

CVPR 2023poster

Exploring dense matching between the current frame and past frames for long-range context modeling, memory-based methods have demonstrated impressive results in video object segmentation (VOS) recently. Nevertheless, due to the lack of instance understanding ability, the above approaches are oftenti…

Cited by 61SourcePDFScholar
2023

i-Code: An Integrative and Composable Multimodal Learning Framework

AAAI 2023technical

Human intelligence is multimodal; we integrate visual, linguistic, and acoustic signals to maintain a holistic worldview. Most current pretraining methods, however, are limited to one or two modalities. We present i-Code, a self-supervised pretraining framework where users may flexibly combine the m…

2022

K-LITE: Learning Transferable Visual Models with External Knowledge

NeurIPS 2022accept

The new generation of state-of-the-art computer vision systems are trained from natural language supervision, ranging from simple object category names to descriptive captions. This form of supervision ensures high generality and usability of the learned visual models, based on the broad concept cov…

2022

OmniVL: One Foundation Model for Image-Language and Video-Language Tasks

NeurIPS 2022accept

This paper presents OmniVL, a new foundation model to support both image-language and video-language tasks using one universal architecture. It adopts a unified transformer-based visual encoder for both image and video inputs, and thus can perform joint image-language and video-language pretraining.…

Cited by 165SourcePDFScholar
2022

REVIVE: Regional Visual Representation Matters in Knowledge-Based Visual Question Answering

NeurIPS 2022accept

This paper revisits visual representation in knowledge-based visual question answering (VQA) and demonstrates that using regional information in a better way can significantly improve the performance. While visual representation is extensively studied in traditional VQA, it is under-explored in kno…

2022

Visual Clues: Bridging Vision and Language Foundations for Image Paragraph Captioning

NeurIPS 2022accept

People say, "A picture is worth a thousand words". Then how can we get the rich information out of the image? We argue that by using visual clues to bridge large pretrained vision foundation models and language models, we can do so without any extra cross-modal training. Thanks to the strong zero-sh…

Cited by 28SourcePDFScholar
2021

A Hypergradient Approach to Robust Regression without Correspondence

ICLR 2021poster

We consider a regression problem, where the correspondence between the input and output data is not available. Such shuffled data are commonly observed in many real world problems. Take flow cytometry as an example: the measuring instruments are unable to preserve the correspondence between the samp…

Cited by 18SourcePDFScholar
2020

Differentiable Top-k with Optimal Transport

NeurIPS 2020poster

Finding the k largest or smallest elements from a collection of scores, i.e., top-k operation, is an important model component widely used in information retrieval, machine learning, and data mining. However, if the top-k operation is implemented in an algorithmic way, e.g., using bubble algorithm,…

2019

A Fast Proximal Point Method for Computing Exact Wasserstein Distance

UAI 2019poster

Wasserstein distance plays increasingly important roles in machine learning, stochastic programming and image processing. Major efforts have been under way to address its high computational complexity, some leading to approximate or regularized variations such as Sinkhorn distance. However, as we wi…

Cited by 224SourcePDFScholar
2019

Meta Learning with Relational Information for Short Sequences

NeurIPS 2019poster

This paper proposes a new meta-learning method -- named HARMLESS (HAwkes Relational Meta Learning method for Short Sequences) for learning heterogeneous point process models from a collection of short event sequence data along with a relational network. Specifically, we propose a hierarchical Bayesi…

2019

On Scalable and Efficient Computation of Large Scale Optimal Transport

ICML 2019oral

Optimal Transport (OT) naturally arises in many machine learning applications, yet the heavy computational burden limits its wide-spread uses. To address the scalability issue, we propose an implicit generative learning-based framework called SPOT (Scalable Push-forward of Optimal Transport). Specif…