← Search

Yangyu Tao

4 accepted papers

2025

Scaling Laws for Floating–Point Quantization Training

ICML 2025poster

Low-precision training is considered an effective strategy for reducing both training and downstream inference costs. Previous scaling laws for precision mainly focus on integer quantization, which pay less attention to the constituents in floating-point (FP) quantization, and thus cannot well fit t…

Cited by 1SourcePDFScholar
2024

Prometheus: Out-of-distribution Fluid Dynamics Modeling with Disentangled Graph ODE

ICML 2024poster

Fluid dynamics modeling has received extensive attention in the machine learning community. Although numerous graph neural network (GNN) approaches have been proposed for this problem, the problem of out-of-distribution (OOD) generalization remains underexplored. In this work, we propose a new large…

Cited by 10SourcePDFScholar
2024

Surge Phenomenon in Optimal Learning Rate and Batch Size Scaling

NeurIPS 2024poster

In current deep learning tasks, Adam-style optimizers—such as Adam, Adagrad, RMSprop, Adafactor, and Lion—have been widely used as alternatives to SGD-style optimizers. These optimizers typically update model parameters using the sign of gradients, resulting in more stable convergence curves. The l…

Cited by 6SourcePDFScholar
2021

Node Dependent Local Smoothing for Scalable Graph Learning

NeurIPS 2021spotlight

Recent works reveal that feature or label smoothing lies at the core of Graph Neural Networks (GNNs). Concretely, they show feature smoothing combined with simple linear regression achieves comparable performance with the carefully designed GNNs, and a simple MLP model with label smoothing of its pr…