NTH

Lumberjack: Better Differentially Private Random Forests through Heavy Hitter Detection in Trees

AuthorsChristian Janos Lebeda, David Erb, Tudor Cebere, Aurélien Bellet

May 25, 2026 2 min read
Watch on YouTube
The one-line take

Lumberjack makes private random forests much more practical by using a clever heavy-hitter pruning trick that keeps deeper trees useful without leaking sensitive data.

Key results

O(log h)
Heavy hitter detector error

The paper’s private hierarchical heavy-hitter detector achieves logarithmic error in tree height h, improving over naive node counting.

1 + floor(log2 h)
Sensitivity reduction

Each data point affects at most 1 + floor(log2 h) threshold queries in the tree-based detector, which is the key privacy/utility improvement.

ε in {0.5, 1, 2, 4, 8}, δ=10^-6
Privacy and evaluation budget

Experiments evaluate Lumberjack under these privacy budgets on Adult and Folktables California classification tasks.

What the paper found

Lumberjack tackles a specific failure mode in differentially private random forests: fully random trees are privacy-friendly but collapse on deep, sparse tabular data, while greedy private splits are brittle and often incorrectly analyzed. The paper’s core novelty is to build very deep random trees first, then privately prune them using a new hierarchical heavy-hitter detector with error scaling O(log h) in tree height h, rather than the O(√h) sensitivity of naive node counting. The detector uses a middle-layer query, monotonicity-based propagation of Heavy/Light labels, and an adaptive Gaussian sparse-histogram thresholding mechanism, yielding a tight joint privacy analysis over the entire forest. In the forest itself, Lumberjack splits features uniformly at random as in Extra Trees, prunes branches whose nodes are not sufficiently populated, and privatizes leaves with either the exponential mechanism for majority vote or Gaussian-noised class proportions under zCDP accounting. Empirically, on Adult and four Folktables California classification tasks, with ε in {0.5, 1, 2, 4, 8} and δ=10^-6, Lumberjack consistently outperforms prior DP random forest baselines including DiffPrivLib, SNR, Smooth Sensitivity, and Suihkonen, often by several accuracy points and sometimes beating a non-private greedy decision tree at moderate privacy. The paper also reports privacy flaws in more than 10 prior DP random-forest papers and shows a lower bound indicating that some threshold growth with tree depth is unavoidable, making Lumberjack’s logarithmic-height detector close to optimal up to a √log h factor.

Original abstract

Random forests are widely used in fields involving sensitive tabular data, but existing approaches to enforcing differential privacy (DP) typically degrade performance to the point of impracticality. In this paper, we introduce Lumberjack, a differentially private random forest algorithm that achieves substantially higher utility by constructing large random decision trees and then applying aggressive, privacy-preserving pruning to retain only sufficiently populated nodes. A key component of our approach is a novel $(\varepsilon,δ)$-DP heavy hitter detection algorithm for hierarchical data, whose error is $O_{\varepsilon,δ}(\sqrt{\log h})$ for trees of height $h$ and may be of independent interest. This favorable scaling enables the use of significantly deeper trees than in prior work, leading to improved expressiveness under privacy constraints. Our empirical evaluation on benchmark datasets shows that Lumberjack consistently outperforms prior DP random forest methods, establishing a new state of the art. In particular, our approach yields substantial improvements in the privacy-utility trade-off for practical privacy budgets. Our findings suggest that carefully designed DP random forests can close much of the utility gap, highlighting a promising and underexplored direction for future research.

Read the original paper

More in Efficient AI

Browse all 55 papers →
01Efficiency

Decoding Looped Transformers Better for (Almost) Free

Weihao Liu, Huangjie Zheng, Tianrong Chen, Rohit Dilip, Richard He Bai, Yizhu Jiao, Yuyang Wang, Ruixiang Zhang

LoopCD turns the partially computed states of looped Transformers into free guidance, improving accuracy while often cutting inference compute nearly in half.

Read analysis
02Efficiency

Scaling Laws for Looped Mixture of Experts

Yanbei Chen, Anirudh Goyal, Raghuraman Krishnamoorthi

This work develops scaling laws that explain how looping and sparse experts can be combined to build more capable models with less training and inference compute.

Read analysis