9

Surrogate Minimization: An Optimization Algorithm for Training Large Neural Networks with Model Parallelism
Optimizing large memory-intensive neural networks requires distributing its layers across multiple GPUs (referred to as model …
The Unsolved Challenges of LLMs in Open-Ended Web Tasks: A Case Study
In this work, we investigate the challenges associated with developing goal-driven AI agents capable of performing open-ended tasks in …
Efficient Dynamics Modeling in Interactive Environments with Koopman Theory
The accurate modeling of dynamics in interactive environments is critical for suc- cessful long-range prediction. Such a capability …
Invariant Causal Set Covering Machines
Rule-based models, such as decision trees, appeal to practitioners due to their interpretable nature. However, the learning algorithms …
Benchmarking Bayesian Causal Discovery Methods for Downstream Treatment Effect Estimation
The practical utility of causality in decision-making is widely recognized, with causal discovery and inference being inherently …
Causal Discovery with Language Models as Imperfect Experts
Understanding the causal relationships that underlie a system is a fundamental prerequisite to accurate decision-making. In this work, …
Explaining Graph Neural Networks Using Interpretable Local Surrogates
We propose an interpretable local surrogate (ILS) method for understanding the predictions of black-box graph models. Explainability …
OC-NMN: Object-centric Compositional Neural Module Network for Generative Visual Analogical Reasoning
Imagination is a crucial aspect of human intelligence that enables us to combine concepts in novel ways and make sense of new …
Workflow discovery in low data regimes
Text-based dialogues are now widely used to solve real-world problems. In cases where solution strategies are already known, they can …
Multilingual Code Retrieval Without Paired Data: A New Benchmark and Experiments
We seek to overcome limitations to code retrieval quality posed by the scarcity of data containing pairs of code snippets and natural …