18 citations · 24 across the 6 of their papers we have counts for
5 papers · 1 filter
Learning Multiple Initial Solutions to Optimization Problems
Elad Sharony, Heng Yang, Tong Che +3
Sequentially solving similar optimization problems under strict runtime constraints is essential for many applications, such as robot control, autonomous driving, and portfolio man…
Parallelized Spatiotemporal Binding
Gautam Singh, Yue Wang, Jiawei Yang +4
While modern best practices advocate for scalable architectures that support long-range interactions, object-centric models are yet to fully embrace these architectures. In particu…
Learning from Teaching Regularization: Generalizable Correlations Should be Easy to Imitate
Can Jin, Tong Che, Hongwu Peng +3
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…
Bayesian Reparameterization of Reward-Conditioned Reinforcement Learning with Energy-based Models
Wenhao Ding, Tong Che, Ding Zhao +1
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…
Foundation Models for Semantic Novelty in Reinforcement Learning
Tarun Gupta, Peter Karkus, Tong Che +2
Effectively exploring the environment is a key challenge in reinforcement learning (RL). We address this challenge by defining a novel intrinsic reward based on a foundation model,…