← Search

Tong Che

20 accepted papers

2026

Learning Multiple Initial Solutions to Optimization Problems

ICRA 2026poster

Sequentially solving similar optimization problems under strict runtime constraints is essential for many applications, such as robot control, autonomous driving, and portfolio management. The performance of local optimization methods in these settings is sensitive to the initial solution: poor init…

2025

LLaMA-Berry: Pairwise Optimization for Olympiad-level Mathematical Reasoning via O1-like Monte Carlo Tree Search

NAACL 2025long

This paper presents LLaMA-Berry, an advanced mathematical reasoning framework to enhance the problem-solving ability of large language models (LLMs). The framework combines Monte Carlo Tree Search with Self-Refine (SR-MCTS) to optimize the reasoning paths and utilizes a pairwise reward model to eval…

Cited by 0SourcePDFScholar
2025

LoR-VP: Low-Rank Visual Prompting for Efficient Vision Model Adaptation

ICLR 2025poster

Visual prompting has gained popularity as a method for adapting pre-trained models to specific tasks, particularly in the realm of parameter-efficient tuning. However, existing visual prompting techniques often pad the prompt parameters around the image, limiting the interaction between the visual p…

2024

EmerNeRF: Emergent Spatial-Temporal Scene Decomposition via Self-Supervision

ICLR 2024poster

We present EmerNeRF, a simple yet powerful approach for learning spatial-temporal representations of dynamic driving scenes. Grounded in neural fields, EmerNeRF simultaneously captures scene geometry, appearance, motion, and semantics via self-bootstrapping. EmerNeRF hinges upon two core components:…

2024

Learning from Teaching Regularization: Generalizable Correlations Should be Easy to Imitate

NeurIPS 2024poster

Generalization remains a central challenge in machine learning. In this work, we propose *Learning from Teaching* (**LoT**), a novel regularization technique for deep neural networks to enhance generalization. Inspired by the human ability to capture concise and abstract patterns, we hypothesize tha…

2024

Parallelized Spatiotemporal Slot Binding for Videos

ICML 2024poster

While modern best practices advocate for scalable architectures that support long-range interactions, object-centric models are yet to fully embrace these architectures. In particular, existing object-centric models for handling sequential inputs, due to their reliance on RNN-based implementation, s…

Cited by 0SourcePDFScholar
2023

Bayesian Reparameterization of Reward-Conditioned Reinforcement Learning with Energy-based Models

ICML 2023poster

Recently, reward-conditioned reinforcement learning (RCRL) has gained popularity due to its simplicity, flexibility, and off-policy nature. However, we will show that current RCRL approaches are fundamentally limited and fail to address two critical challenges of RCRL -- improving generalization on…

Cited by 5SourcePDFScholar
2023

Guided Conditional Diffusion for Controllable Traffic Simulation

ICRA 2023poster

Controllable and realistic traffic simulation is critical for developing and verifying autonomous vehicles. Typical heuristic-based traffic models offer flexible control to make vehicles follow specific trajectories and traffic rules. On the other hand, data-driven approaches generate realistic and…

Cited by 167SourcecodeScholar
2023

Robust and Controllable Object-Centric Learning through Energy-based Models

ICLR 2023poster

Humans are remarkably good at understanding and reasoning about complex visual scenes. The capability of decomposing low-level observations into discrete objects allows us to build a grounded abstract representation and identify the compositional structure of the world. Thus it is a crucial step for…

Cited by 12SourcePDFScholar
2023

Sparse Mixture-of-Experts are Domain Generalizable Learners

ICLR 2023top-5%

Human visual perception can easily generalize to out-of-distributed visual data, which is far beyond the capability of modern machine learning models. Domain generalization (DG) aims to close this gap, with existing DG methods mainly focusing on the loss function design. In this paper, we propose to…

2022

SPE: Symmetrical Prompt Enhancement for Fact Probing

EMNLP 2022main

Pretrained language models (PLMs) have been shown to accumulate factual knowledge during pretraining (Petroni et al. 2019). Recent works probe PLMs for the extent of this knowledge through prompts either in discrete or continuous forms. However, these methods do not consider symmetry of the task: ob…

Cited by 8SourcePDFScholar
2021

Deep Verifier Networks: Verification of Deep Discriminative Models with Deep Generative Models

AAAI 2021technical

AI Safety is a major concern in many deep learning applications such as autonomous driving. Given a trained deep learning model, an important natural problem is how to reliably verify the model's prediction. In this paper, we propose a novel framework --- deep verifier networks (DVN) to detect unrel…

Cited by 67SourcePDFScholar
2021

Energy-Based Open-World Uncertainty Modeling for Confidence Calibration

ICCV 2021poster

Confidence calibration is of great importance to ensure the reliability of decisions made by machine learning systems. However, discriminative classifiers based on deep neural networks are often criticized for producing overconfident predictions that fail to reflect the true correctness likelihood o…

Cited by 70PDFScholar
2020

AUTO3D: Novel view synthesis through unsupervisely learned variational viewpoint and global 3D representation

ECCV 2020poster

This paper targets on learning-based novel view synthesis from a single or limited 2D images without the pose supervision. In the viewer-centered coordinates, we construct an end-to-end trainable conditional variational framework to disentangle the unsupervisely learned relative-pose/rotation and im…

Cited by 26SourcePDFScholar
2020

Your GAN is Secretly an Energy-based Model and You Should Use Discriminator Driven Latent Sampling

NeurIPS 2020poster

We show that the sum of the implicit generator log-density $\log p_g$ of a GAN with the logit score of the discriminator defines an energy function which yields the true data density when the generator is imperfect but the discriminator is optimal, thus making it possible to improve on the typical g…

Cited by 147SourcePDFScholar
2018

MetaGAN: An Adversarial Approach to Few-Shot Learning

NeurIPS 2018poster

In this paper, we propose a conceptually simple and general framework called MetaGAN for few-shot learning problems. Most state-of-the-art few-shot classification models can be integrated with MetaGAN in a principled and straightforward way. By introducing an adversarial generator conditioned on tas…

Cited by 724SourcePDFScholar
2018

Residual Connections Encourage Iterative Inference

ICLR 2018poster

Residual networks (Resnets) have become a prominent architecture in deep learning. However, a comprehensive understanding of Resnets is still a topic of ongoing research. A recent view argues that Resnets perform iterative refinement of features. We attempt to further expose properties of this aspec…

Cited by 174SourcePDFScholar
2016

Architectural Complexity Measures of Recurrent Neural Networks

NeurIPS 2016poster

In this paper, we systematically analyze the connecting architectures of recurrent neural networks (RNNs). Our main contribution is twofold: first, we present a rigorous graph-theoretic framework describing the connecting architectures of RNNs in general. Second, we propose three architecture comple…