← Search

Yedi Zhang

7 accepted papers

2026

Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architectures

ICLR 2026poster

Neural networks trained with gradient descent often learn solutions of increasing complexity over time, a phenomenon known as simplicity bias. Despite being widely observed across architectures, existing theoretical treatments lack a unifying framework. We present a theoretical framework that explai…

Cited by 0SourceScholar
2026

Towards Stealthy and Effective Backdoor Attacks on Lane Detection: A Naturalistic Data Poisoning Approach

CVPR 2026

Deep learning-based lane detection (LD) plays a critical role in autonomous driving and advanced driver assistance systems. However, its vulnerability to backdoor attacks presents a significant security concern. Existing backdoor attack methods on LD often exhibit limited practical utility due to th

Cited by 0SourceScholar
2025

Position: Trustworthy AI Agents Require the Integration of Large Language Models and Formal Methods

ICML 2025poster

Large Language Models (LLMs) have emerged as a transformative AI paradigm, profoundly influencing broad aspects of daily life. Despite their remarkable performance, LLMs exhibit a fundamental limitation: hallucination—the tendency to produce misleading outputs that appear plausible. This inherent…

Cited by 0SourcePDFScholar
2025

Training Dynamics of In-Context Learning in Linear Attention

ICML 2025spotlight

While attention-based models have demonstrated the remarkable ability of in-context learning (ICL), the theoretical understanding of how these models acquired this ability through gradient descent training is still preliminary. Towards answering this question, we study the gradient descent dynamics…

Cited by 1SourcePDFScholar
2024

Towards Efficient Verification of Quantized Neural Networks

AAAI 2024technical

Quantization replaces floating point arithmetic with integer arithmetic in deep neural network models, providing more efficient on-device inference with less power and memory. In this work, we propose a framework for formally verifying the properties of quantized neural networks. Our baseline techni…

2024

Understanding Unimodal Bias in Multimodal Deep Linear Networks

ICML 2024poster

Using multiple input streams simultaneously to train multimodal neural networks is intuitively advantageous but practically challenging. A key challenge is unimodal bias, where a network overly relies on one modality and ignores others during joint training. We develop a theory of unimodal bias with…