activity
20182025
most citedOnly Train Once: A One-Shot Neural Network Training And Pruning Framework

43 citations · 54 across the 11 of their papers we have counts for

collaborators

11 papers

cs.LG2025

Topic Analysis with Side Information: A Neural-Augmented LDA Approach

Biyi Fang, Truong Vo, Kripa Rajshekhar +1

Traditional topic models such as Latent Dirichlet Allocation (LDA) have been widely used to uncover latent structures in text corpora, but they often struggle to integrate auxiliar…

cs.CL2025

TrInk: Ink Generation with Transformer Network

Zezhong Jin, Shubhang Desai, Xu Chen +8

In this paper, we propose TrInk, a Transformer-based model for ink generation, which effectively captures global dependencies. To better facilitate the alignment between the input…

cs.LG2025

Tricks and Plug-ins for Gradient Boosting with Transformers

Biyi Fang, Truong Vo, Jean Utke +1

Transformer architectures dominate modern NLP but often demand heavy computational resources and intricate hyperparameter tuning. To mitigate these challenges, we propose a novel f…

stat.ML2025

Tricks and Plug-ins for Gradient Boosting in Image Classification

Biyi Fang, Truong Vo, Jean Utke +1

Convolutional Neural Networks (CNNs) have achieved remarkable success across a wide range of machine learning tasks by leveraging hierarchical feature learning through deep archite…

cs.CL2022

SLATE: A Sequence Labeling Approach for Task Extraction from Free-form Inked Content

Apurva Gandhi, Ryan Serrao, Biyi Fang +8

We present SLATE, a sequence labeling approach for extracting tasks from free-form content such as digitally handwritten (or "inked") notes on a virtual whiteboard. Our approach al…

cs.LG2022

Topic Analysis for Text with Side Data

Biyi Fang, Kripa Rajshekhar, Diego Klabjan

Although latent factor models (e.g., matrix factorization) obtain good performance in predictions, they suffer from several problems including cold-start, non-transparency, and sub…

cs.LG2022

Tricks and Plugins to GBM on Images and Sequences

Biyi Fang, Jean Utke, Diego Klabjan

Convolutional neural networks (CNNs) and transformers, which are composed of multiple processing layers and blocks to learn the representations of data with multiple abstract level…

cs.LG2021★ 43 cited

Only Train Once: A One-Shot Neural Network Training And Pruning Framework

Tianyi Chen, Bo Ji, Tianyu Ding +7

Structured pruning is a commonly used technique in deploying deep neural networks (DNNs) onto resource-constrained devices. However, the existing pruning methods are usually heuris…

cs.LG2020★ 6 cited

Neural Network Compression Via Sparse Optimization

Tianyi Chen, Bo Ji, Yixin Shi +4

The compression of deep neural networks (DNNs) to reduce inference cost becomes increasingly important to meet realistic deployment requirements of various applications. There have…

cs.LG2019★ 4 cited

Convergence Analyses of Online ADAM Algorithm in Convex Setting and Two-Layer ReLU Neural Network

Biyi Fang, Diego Klabjan

Nowadays, online learning is an appealing learning paradigm, which is of great interest in practice due to the recent emergence of large scale applications such as online advertisi…

stat.ML2018★ 1 cited

A Stochastic Large-scale Machine Learning Algorithm for Distributed Features and Observations

Biyi Fang, Diego Klabjan

As the size of modern data sets exceeds the disk and memory capacities of a single computer, machine learning practitioners have resorted to parallel and distributed computing. Giv…