← Search

Yankai Cao

7 accepted papers

2026

ReFTA: Breaking the Weight Reconstruction Bottleneck in Tensorized Parameter-Efficient Fine-Tuning

CVPR 2026

Tensor-based methods have attracted growing interest due to their ability to reduce trainable parameters and offer advantages over matrix-based approaches in parameter-efficient fine-tuning (e.g., LoRA and PiSSA), particularly in capturing inter-layer correlations. However, directly applying tensor

Cited by 0SourcecodeScholar
2025

AdaMSS: Adaptive Multi-Subspace Approach for Parameter-Efficient Fine-Tuning

NeurIPS 2025poster

In this paper, we propose AdaMSS, an adaptive multi-subspace approach for parameter-efficient fine-tuning of large models. Unlike traditional parameter-efficient fine-tuning methods that operate within a large single subspace of the network weights, AdaMSS leverages subspace segmentation to obtain…

Cited by 0SourcecodeScholar
2025

Differentiable Decision Tree via "ReLU+Argmin" Reformulation

NeurIPS 2025spotlight

Decision tree, despite its unmatched interpretability and lightweight structure, faces two key issues that limit its broader applicability: non-differentiability and low testing accuracy. This study addresses these issues by developing a differentiable oblique tree that optimizes the entire tree us…

Cited by 0SourcecodeScholar
2022

A Scalable Deterministic Global Optimization Algorithm for Training Optimal Decision Tree

NeurIPS 2022accept

The training of optimal decision tree via mixed-integer programming (MIP) has attracted much attention in recent literature. However, for large datasets, state-of-the-art approaches struggle to solve the optimal decision tree training problems to a provable global optimal solution within a reasonabl…

Cited by 9SourcePDFScholar
2021

A Scalable Deterministic Global Optimization Algorithm for Clustering Problems

ICML 2021spotlight

The minimum sum-of-squares clustering (MSSC) task, which can be treated as a Mixed Integer Second Order Cone Programming (MISOCP) problem, is rarely investigated in the literature through deterministic optimization to find its global optimal value. In this paper, we modelled the MSSC task as a two-s…

Cited by 7SourcePDFScholar