← Search

Zekai Wang

7 accepted papers

2025

In-Context Learning Enables Robot Action Prediction in LLMs

ICRA 2025

Recently, Large Language Models (LLMs) have achieved remarkable success using in-context learning (ICL) in the language domain. However, leveraging the ICL capabilities within LLMs to directly predict robot actions remains largely unexplored. In this paper, we introduce RoboPrompt, a frame-work that

Cited by 19SourcecodeScholar
2024

DRF: Improving Certified Robustness via Distributional Robustness Framework

AAAI 2024technical

Randomized smoothing (RS) has provided state-of-the-art (SOTA) certified robustness against adversarial perturbations for large neural networks. Among studies in this field, methods based on adversarial training (AT) achieve remarkably robust performance by applying adversarial examples to construct…

Cited by 2SourcePDFScholar
2023

Better Diffusion Models Further Improve Adversarial Training

ICML 2023poster

It has been recognized that the data generated by the denoising diffusion probabilistic model (DDPM) improves adversarial training. After two years of rapid development in diffusion models, a question naturally arises: can better diffusion models further improve adversarial training? This paper give…

2022

MetaWeighting: Learning to Weight Tasks in Multi-Task Learning

ACL 2022findings

Task weighting, which assigns weights on the including tasks during training, significantly matters the performance of Multi-task Learning (MTL); thus, recently, there has been an explosive interest in it. However, existing task weighting methods assign weights only based on the training loss, while…

Cited by 27SourcePDFScholar
2021

BanditMTL: Bandit-based Multi-task Learning for Text Classification

ACL 2021long

Task variance regularization, which can be used to improve the generalization of Multi-task Learning (MTL) models, remains unexplored in multi-task text classification. Accordingly, to fill this gap, this paper investigates how the task might be effectively regularized, and consequently proposes a m…

Cited by 19SourcePDFScholar