← Search

Yun Yue

6 accepted papers

2025

EDiT: A Local-SGD-Based Efficient Distributed Training Method for Large Language Models

ICLR 2025poster

Distributed training methods are crucial for large language models (LLMs). However, existing distributed training methods often suffer from communication bottlenecks, stragglers, and limited elasticity, particularly in heterogeneous or large-scale environments. Local SGD methods have been proposed t…

2024

Integer Is Enough: When Vertical Federated Learning Meets Rounding

AAAI 2024technical

Vertical Federated Learning (VFL) is a solution increasingly used by companies with the same user group but differing features, enabling them to collaboratively train a machine learning model. VFL ensures that clients exchange intermediate results extracted by their local models, without sharing ra…

Cited by 2SourcePDFScholar
2023

AGD: an Auto-switchable Optimizer using Stepwise Gradient Difference for Preconditioning Matrix

NeurIPS 2023poster

Adaptive optimizers, such as Adam, have achieved remarkable success in deep learning. A key component of these optimizers is the so-called preconditioning matrix, providing enhanced gradient information and regulating the step size of each gradient direction. In this paper, we propose a novel approa…

2023

Hyperbolic Chamfer Distance for Point Cloud Completion

ICCV 2023poster

Chamfer distance (CD) is a standard metric to measure the shape dissimilarity between point clouds in point cloud completion, as well as a loss function for (deep) learning. However, it is well known that CD is vulnerable to outliers, leading to the drift towards suboptimal models. In contrast to th…

Cited by 43PDFScholar
2023

InfoCD: A Contrastive Chamfer Distance Loss for Point Cloud Completion

NeurIPS 2023poster

A point cloud is a discrete set of data points sampled from a 3D geometric surface. Chamfer distance (CD) is a popular metric and training loss to measure the distances between point clouds, but also well known to be sensitive to outliers. To address this issue, in this paper we propose InfoCD, a no…

2021

SBO-RNN: Reformulating Recurrent Neural Networks via Stochastic Bilevel Optimization

NeurIPS 2021poster

In this paper we consider the training stability of recurrent neural networks (RNNs) and propose a family of RNNs, namely SBO-RNN, that can be formulated using stochastic bilevel optimization (SBO). With the help of stochastic gradient descent (SGD), we manage to convert the SBO problem into an RNN…