Search Results for author: Yujia Bao

Found 9 papers, 8 papers with code

Sample, estimate, aggregate: A recipe for causal discovery foundation models

1 code implementation2 Feb 2024 Menghua Wu, Yujia Bao, Regina Barzilay, Tommi Jaakkola

Causal discovery, the task of inferring causal structure from data, promises to accelerate scientific research, inform policy making, and more.

Causal Discovery

Channel Vision Transformers: An Image Is Worth 1 x 16 x 16 Words

1 code implementation28 Sep 2023 Yujia Bao, Srinivasan Sivanandan, Theofanis Karaletsos

We evaluate the performance of ChannelViT on ImageNet, JUMP-CP (microscopy cell imaging), and So2Sat (satellite imaging).

Contextual Vision Transformers for Robust Representation Learning

1 code implementation30 May 2023 Yujia Bao, Theofanis Karaletsos

Additionally, we introduce a context inference network to predict such tokens on-the-fly, given a batch of samples from the group.

In-Context Learning Out-of-Distribution Generalization +1

Learning Stable Classifiers by Transferring Unstable Features

1 code implementation15 Jun 2021 Yujia Bao, Shiyu Chang, Regina Barzilay

Empirical results demonstrate that our algorithm is able to maintain robustness on the target task for both synthetically generated environments and real-world environments.

Transfer Learning

Predict then Interpolate: A Simple Algorithm to Learn Stable Classifiers

1 code implementation26 May 2021 Yujia Bao, Shiyu Chang, Regina Barzilay

In this work, we prove that by interpolating the distributions of the correct predictions and the wrong predictions, we can uncover an oracle distribution where the unstable correlation vanishes.

Image Classification text-classification +1

Deriving Machine Attention from Human Rationales

3 code implementations EMNLP 2018 Yujia Bao, Shiyu Chang, Mo Yu, Regina Barzilay

Attention-based models are successful when trained on large amounts of data.

Cannot find the paper you are looking for? You can Submit a new open access paper.