Research

Students

I'm glad and lucky to advise and collaborate with:

Mr. Shangyu Wu

Miss Yu Mao

Patents

Artificial neural network configuration and deployment
On the Ordered Dropout and Fully Nested Neural Network in “Fully Nested Neural Network for Adaptive Compression and Quantization”
Tei-Wei Kuo, Antoni Bert Chan, Chun Xue, Yufei Cui, Ziquan Liu, Wuguannan Yao, Qiao Li
U.S. Patent Application No. 17/000,612.

Publications

Improving Natural Language Understanding with Computation-Efficient Retrieval Representation Fusion
Shangyu Wu, Ying Xiong, Yufei Cui (corresponding and co-first), Xue Liu, Buzhou Tang, Tei-Wei Kuo, Chun Jason Xue
NeurIPS 2023 3rd Workshop on Efficient Natural Language and Speech Processing, ENLSP-NeurIPS 2023
Twelfth International Conference on Learning Representations, ICLR 2024

Retrieval-Augmented Multiple Instance Learning
Yufei Cui, Ziquan Liu, Yixin Chen, Yuchen Lu, Xinyue Yu, Xue Liu, Tei-Wei Kuo, Miguel R. D. Rodrigues, Chun Jason Xue, Antoni B. Chan
Conference on Neural Information Processing Systems, NeurIPS 2023

Faster and stronger Lossless Compression with Optimized Autoregressive framework
Yu Mao, Jingzong Li, Yufei Cui, Chun Jason Xue
Design Automation Conference, DAC 2023

Bayes-MIL: A New Probabilistic Perspective on Attention-based Multiple Instance Learning for Whole Slide Images
Yufei Cui, Ziquan Liu, Xiangyu Liu, Xue Liu, Cong Wang, Tei-Wei Kuo, Chun Jason Xue, Antoni B. Chan
The International Conference on Learning Representations, ICLR 2023

Variational Nested Dropout
Yufei Cui, Yu Mao, Ziquan Liu, Qiao Li, Antoni B. Chan, Xue Liu, Tei-Wei Kuo, Chun Jason Xue
IEEE Transactions on Pattern Analysis and Machine Intelligence, TPAMI

Precise Augmentation and Counting of Helicobacter Pylori in Histology Image
Yufei Cui, Yixin Chen, Zhifeng Shuai, Fang Peng, Yanbo Lv, Luoning Zheng, Xue Liu, Antoni B. Chan, Tei-Wei Kuo, Chun Jason Xue
Medical Imaging meets NeurIPS, MED-NEURIPS 2022

Bits-Ensemble: Towards Light-Weight Robust Deep Ensemble by Bits-Sharing
Yufei Cui, Shangyu Wu, Qiao Li, Antoni B. Chan, Wei-Wei Kuo and Chun Jason Xue
Proceedings of the 2022 International Conference on Compilers, Architectures, and Synthesis for Embedded Systems, CASES 2022, to appear
IEEE Transactions on Computer Aided Design of Integrated Circuits & Systems, TCAD

Resolving the Reliability Issues of Open Blocks for 3D NAND Flash: Observations and Strategies
Qiao Li, Min Ye, Yufei Cui, Tianyu Ren, Tei-Wei Kuo, Chun Jason Xue
Proceedings of the 2022 International Conference on Hardware/Software Codesign and System Synthesis, CODES+ISSS 2022, to appear
IEEE Transactions on Computer Aided Design of Integrated Circuits & Systems, TCAD

Accelerating General-purpose Lossless Compression via Simple and Scalable Parameterization
Yu Mao, Yufei Cui (Corresponding), Tei-Wei Kuo and Chun Jason Xue
The 30th ACM International Conference on Multimedia, ACM MM 2022

NFL: Robust Learned Index via Distribution Transformation
Shangyu Wu, Yufei Cui (Corresponding), Jinghuan Yu, Xuan Sun, Tei-Wei Kuo and Chun Jason Xue
The 48th International Conference on Very Large Databases, VLDB 2022

TRACE: A Fast Transformer-based General-Purpose Lossless Compressor
Yu Mao, Yufei Cui (Corresponding), Tei-Wei Kuo and Chun Jason Xue
The Web Conference, WWW 2022

CacheSifter: Sifting Cache Files for Boosted Mobile Performance and Lifetime
Yu Liang, Riwei Pan, Tianyu Ren, Yufei Cui, Rachata Ausavarungnirun, Xianzhang Chen, Changlong Li, Tei-Wei Kuo, Chun Jason Xue
The 20th USENIX Conference on File and Storage Technologies, FAST 2022.

Improve Generalization and Robustness of Neural Networks via Weight Scale Shifting Invariant Regularizations
Ziquan Liu, Yufei Cui, Antoni B. Chan.
ICML Workshop on Adversarial Machine Learning, ICMLW 2021.

FlashEmbedding: Storing Embedding Tables in SSD for Large-Scale Recommender Systems
Hu Wan, Xuan Sun, Yufei Cui, Chia-Lin Yang, Tei-Wei Kuo, Chun Jason Xue.
The ACM Asia-Pacific Workshop on Systems, ApSys 2021, to appear.

Online Rare Category Identification and Data Diversification for Edge Computing
Yufei Cui, Qiao Li, Chun Jason Xue, Tei-Wei Kuo.
IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems (TCAD).

Bayesian Nested Neural Networks for Uncertainty Calibration and Adaptive Compression
Yufei Cui, Ziquan Liu, Qiao Li, Antoni B. Chan, Chun Jason Xue.
IEEE/CVF Conference on Computer Vision and Pattern Recognition, CVPR 2021.

Fully Nested Neural Network for Adaptive Compression and Quantization
Yufei Cui, Ziquan Liu, Wuguannan Yao, Qiao Li, Antoni B. Chan, Tei-Wei Kuo, Chun Jason Xue.
International Joint Conference on Artificial Intelligence, IJCAI 2020.

Accelerating Monte Carlo Bayesian Prediction via Approximating Predictive Uncertainty Over the Simplex
Yufei Cui, Wuguannan Yao, Qiao Li, Antoni B. Chan, Chun Jason Xue.
IEEE Transactions on Neural Networks and Learning Systems, TNNLS.

Pruning Deep Reinforcement Learning for Dual User Experience and Storage Lifetime Improvement on Mobile Devices
Chao Wu, Yufei Cui (Co-first author), Cheng Ji, Tei-Wei Kuo, Chun Jason Xue.
IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems (TCAD).
International Conference on Embedded Software, EMSOFT 2020.

Exploiting Asymmetric Errors for LDPC Decoding Optimization on 3D NAND Flash Memory
Qiao Li, Liang Shi, Yufei Cui, Chun Jason Xue.
Transaction on Computers (TC).

Shaving Retries with Sentinels for Fast Read over High-Density 3D Flash
Qiao Li, Min Ye, Yufei Cui, Liang Shi, Xiaoqiang Li, Tei-Wei Kuo, Chun Jason Xue.
IEEE/ACM International Symposium on Microarchitecture, MICRO 2020.

Online Rare Category Detection for Edge Computing
Yufei Cui, Qiao Li, Sarana Nutanong, Chun Jason Xue.
Design, Automation & Test in Europe Conference & Exhibition, DATE 2019.

A Hardware-Accelerated Solution for Hierarchical Index-Based Merge-Join
Zimeng Zhou, Chenyun Yu, Sarana Nutanong, Yufei Cui, Chenchen Fu, Chun Jason Xue.
IEEE Transactions on Knowledge and Data Engineering (TKDE).
International Conference on Data Engineering, ICDE 2019.

Sentinel Cells Enabled Fast Read for NAND Flash
Qiao Li, Min Ye, Yufei Cui, Liang Shi, Xiaoqiang Li, Chun Jason Xue.
USENIX Workshop on Hot Topics in Storage and File Systems, HotStorage 2019.