← Search

Dehong Xu

7 accepted papers

2025

Latent Thought Models with Variational Bayes Inference-Time Computation

ICML 2025poster

We propose a novel class of language models, Latent Thought Models (LTMs), which incorporate explicit latent thought vectors that follow an explicit prior model in latent space. These latent thought vectors guide the autoregressive generation of ground tokens through a Transformer decoder. Training…

2025

Multi-Modal Multi-Task Unified Embedding Model (M3T-UEM): A Task-Adaptive Representation Learning Framework

ICCV 2025poster

We present Multi-Modal Multi-Task Unified Embedding Model (M3T-UEM), a framework that advances vision-language matching and retrieval by leveraging a large language model (LLM) backbone. While concurrent LLM-based approaches like VLM2VEC, MM-Embed, NV-Embed, and MM-GEM have demonstrated impressive c…

2025

On Conformal Isometry of Grid Cells: Learning Distance-Preserving Position Embedding

ICLR 2025oral

This paper investigates the conformal isometry hypothesis as a potential explanation for the hexagonal periodic patterns in grid cell response maps. We posit that grid cell activities form a high-dimensional vector in neural space, encoding the agent's position in 2D physical space. As the agent mov…

Cited by 0SourcePDFScholar
2025

Place Cells as Multi-Scale Position Embeddings: Random Walk Transition Kernels for Path Planning

NeurIPS 2025poster

The hippocampus supports spatial navigation by encoding cognitive maps through collective place cell activity. We model the place cell population as non-negative spatial embeddings derived from the spectral decomposition of multi-step random walk transition kernels. In this framework, inner product…

Cited by 0SourceScholar
2024

Aligning Large Language Models via Fine-grained Supervision

ACL 2024short

Pre-trained large-scale language models (LLMs) excel at producing coherent articles, yet their outputs may be untruthful, toxic, or fail to align with user expectations. Current approaches focus on using reinforcement learning with human feedback (RLHF) to improve model alignment, which works by tra…

Cited by 2SourcePDFScholar
2024

Latent Plan Transformer for Trajectory Abstraction: Planning as Latent Space Inference

NeurIPS 2024poster

In tasks aiming for long-term returns, planning becomes essential. We study generative modeling for planning with datasets repurposed from offline reinforcement learning. Specifically, we identify temporal consistency in the absence of step-wise rewards as one key technical challenge. We introduce t…

2023

Diverse and Faithful Knowledge-Grounded Dialogue Generation via Sequential Posterior Inference

ICML 2023poster

The capability to generate responses with diversity and faithfulness using factual knowledge is paramount for creating a human-like, trustworthy dialogue system. Common strategies either adopt a two-step paradigm, which optimizes knowledge selection and response generation separately, and may overlo…