← Search

David W Zhang

8 accepted papers

2024

CodeIt: Self-Improving Language Models with Prioritized Hindsight Replay

ICML 2024poster

Large language models are increasingly solving tasks that are commonly believed to require human-level reasoning ability. However, these models still perform very poorly on benchmarks of general intelligence such as the Abstraction and Reasoning Corpus (ARC). In this paper, we approach the ARC as a…

2024

Graph Neural Networks for Learning Equivariant Representations of Neural Networks

ICLR 2024oral

Neural networks that process the parameters of other neural networks find applications in domains as diverse as classifying implicit neural representations, generating neural network weights, and predicting generalization errors. However, existing approaches either overlook the inherent permutation…

2024

Improved Generalization of Weight Space Networks via Augmentations

ICML 2024poster

Learning in deep weight spaces (DWS), where neural networks process the weights of other neural networks, is an emerging research direction, with applications to 2D and 3D neural fields (INRs, NeRFs), as well as making inferences about other types of neural networks. Unfortunately, weight space mode…

2023

Self-Guided Diffusion Models

CVPR 2023poster

Diffusion models have demonstrated remarkable progress in image generation quality, especially when guidance is used to control the generative process. However, guidance requires a large amount of image-annotation pairs for training and is thus dependent on their availability and correctness. In thi…

2023

Unlocking Slot Attention by Changing Optimal Transport Costs

ICML 2023poster

Slot attention is a powerful method for object-centric modeling in images and videos. However, its set-equivariance limits its ability to handle videos with a dynamic number of objects because it cannot break ties. To overcome this limitation, we first establish a connection between slot attention a…

2022

Multiset-Equivariant Set Prediction with Approximate Implicit Differentiation

ICLR 2022poster

Most set prediction models in deep learning use set-equivariant operations, but they actually operate on multisets. We show that set-equivariant functions cannot represent certain functions on multisets, so we introduce the more appropriate notion of multiset-equivariance. We identify that the exist…

2021

Set Prediction without Imposing Structure as Conditional Density Estimation

ICLR 2021poster

Set prediction is about learning to predict a collection of unordered variables with unknown interrelations. Training such models with set losses imposes the structure of a metric space over sets. We focus on stochastic and underdefined cases, where an incorrectly chosen loss function leads to impla…