AIAny
AI Train2016
Icon for item

LightGBM

Trains gradient-boosted decision trees for classification, ranking, and large-scale tabular ML with lower memory use and faster training. GOSS and EFB help it handle high-dimensional sparse data on CPU, GPU, and distributed setups.

Introduction

Gradient-boosted trees have stayed relevant because many production datasets are still tabular, messy, and expensive to train at scale. The interesting move here is not inventing a new model family, but making a strong old one practical on wider, sparser, larger datasets.

What Sets It Apart
  • Uses Gradient-based One-Side Sampling to train trees from the most informative instances, so large datasets can be processed without scanning every row with equal cost.
  • Bundles mutually exclusive sparse features, which matters for high-dimensional one-hot and categorical-style data where naive split search becomes memory-heavy.
  • Supports CPU, GPU, parallel, and distributed training paths, making it useful from local experimentation to larger ranking or classification workloads.
  • Its long history in machine learning competitions gives practitioners a mature baseline to compare against neural tabular methods and AutoML systems.
Where It Fits

It sits in the same practical family as XGBoost and CatBoost: fast supervised learning for structured data, ranking, and regression. Its strongest niche is when training speed, memory pressure, and sparse feature handling matter more than end-to-end deep representation learning.

Fit and Tradeoffs

Great fit if you need a battle-tested tree boosting engine for tabular data, learning-to-rank, or large-scale feature engineering pipelines. Look elsewhere if your core problem is unstructured media, sequence modeling, or a workflow where interpretability and calibration matter more than leaderboard-style predictive performance.

Information

  • Websitelightgbm.readthedocs.io
  • OrganizationsMicrosoft Research, Microsoft
  • AuthorsMicrosoft (originally Microsoft Research)
  • Published date2016/08/05

Categories

More Items

GitHub
AI Train2019

Train and experiment with multi-billion to trillion-parameter transformer models on large GPU clusters using GPU-optimized building blocks and reference training scripts; offers advanced parallelism and mixed-precision support for research teams and ML engineers.

GitHub
AI Train2026

Provides a one-command CLI to fine-tune and post-train LLMs, with layer streaming that lets an 8B model be fine-tuned on a 4 GB laptop GPU. Auto-configures quantization, LoRA adapters, batching and evaluation gates, and supports export and serving workflows.

GitHub
AI Train2025

A PyTorch DTensor-native SPMD library for training and fine-tuning LLMs, VLMs, diffusion and retrieval models. Integrates with Hugging Face for day-0 model support, provides YAML-driven recipes, DTensor/FSDP2 parallelism and NVIDIA-optimized kernels (Transformer Engine, DeepEP, FlexAttn).