NUS-HPC-AI-Lab / Helen
The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"
☆15Updated 10 months ago
Alternatives and similar repositories for Helen:
Users that are interested in Helen are comparing it to the libraries listed below
- [NeurIPS 2024] Official code of $\beta$-DPO: Direct Preference Optimization with Dynamic $\beta$☆38Updated 2 months ago
- ☆119Updated 6 months ago
- ☆49Updated 8 months ago
- This pytorch package implements PLATON: Pruning Large Transformer Models with Upper Confidence Bound of Weight Importance (ICML 2022).☆43Updated 2 years ago
- [NeurIPS 2023] Model-enhanced Vector Index☆25Updated 8 months ago
- Less is More: Task-aware Layer-wise Distillation for Language Model Compression (ICML2023)☆32Updated last year
- Official implementation for Yuan & Liu & Zhong et al., KV Cache Compression, But What Must We Give in Return? A Comprehensive Benchmark o…☆61Updated 3 weeks ago
- ☆49Updated last year
- ☆73Updated last year
- Learning to Tokenize for Generative Retrieval (NeurIPS 2023)☆52Updated 2 months ago
- PyTorch implementation of paper "Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline".☆81Updated last year
- [AAAI 2024] Fluctuation-based Adaptive Structured Pruning for Large Language Models☆41Updated last year
- ☆16Updated last month
- Ouroboros: Speculative Decoding with Large Model Enhanced Drafting (EMNLP 2024 main)☆85Updated 3 months ago
- This package implements THOR: Transformer with Stochastic Experts.☆61Updated 3 years ago
- [ICLR 2024 Spotlight] Code for the paper "Merge, Then Compress: Demystify Efficient SMoE with Hints from Its Routing Policy"☆70Updated 7 months ago
- Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding (EMNLP 2023 Long)☆56Updated 3 months ago
- Reproduction of "RLCD Reinforcement Learning from Contrast Distillation for Language Model Alignment☆64Updated last year
- ☆46Updated 2 months ago
- The official implementation of paper: SimLayerKV: A Simple Framework for Layer-Level KV Cache Reduction.☆42Updated 3 months ago
- ☆20Updated 6 months ago
- [NeurIPS 2024] Fast Best-of-N Decoding via Speculative Rejection☆38Updated 2 months ago
- Official implementation of ICML 2024 paper "ExCP: Extreme LLM Checkpoint Compression via Weight-Momentum Joint Shrinking".☆44Updated 6 months ago
- SepLLM: Accelerate Large Language Models by Compressing One Segment into One Separator☆42Updated last month
- [SIGIR'24] Generative Retrieval as Multi-Vector Dense Retrieval☆29Updated 3 months ago
- ☆30Updated 4 months ago
- The official code for paper "parallel speculative decoding with adaptive draft length."☆32Updated 4 months ago
- A Closer Look into Mixture-of-Experts in Large Language Models☆41Updated 5 months ago
- This repo contains the source code for: Model Tells You What to Discard: Adaptive KV Cache Compression for LLMs☆32Updated 5 months ago
- This PyTorch package implements MoEBERT: from BERT to Mixture-of-Experts via Importance-Guided Adaptation (NAACL 2022).☆97Updated 2 years ago