DNNR: Differential Nearest Neighbors Regression
Youssef Nader, Leon Sixt, Tim Landgraf
Abstract
K-nearest neighbors (KNN) is one of the earli-est and most established algorithms in machine learning. For regression tasks, KNN averages the targets within a neighborhood which poses a number of challenges: the neighborhood definition is crucial for the predictive performance as neighbors might be selected based on uninformative features, and averaging does not account for how the function changes locally. We propose a novel method called Differential Nearest Neighbors Regression (DNNR) that addresses both issues simul-taneously: during training, DNNR estimates local gradients to scale the features; during inference, it performs an n-th order Taylor approximation using estimated gradients. In a large-scale evaluation on over 250 datasets, we find that DNNR performs comparably to state-of-the-art gradient boosting methods and MLPs while maintaining the simplicity and transparency of KNN. This allows us to derive theoretical error bounds and inspect failures. In times that call for transparency of ML models, DNNR provides a good balance between performance and interpretability. 2
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 9e46e5fb-c45f-4029-8a3d-c3168d529b98Cited by top-tier papers6
- TabR: Tabular Deep Learning Meets Nearest NeighborsYury Gorishniy, Ivan Rubachev, Nikolay Kartashev, Daniil Shlenskii et al.ICLR 2024 · 78 citations
- Retrieval & Fine-Tuning for In-Context Tabular ModelsValentin Thomas, Junwei Ma, Rasa Hosseinzadeh, Keyvan Golestan et al.NeurIPS 2024 · 75 citations
- A Closer Look at TabPFN v2: Understanding Its Strengths and Extending Its CapabilitiesHan-Jia Ye, Si-Yang Liu, Wei-Lun ChaoNeurIPS 2025 · 52 citations
- Elliptical AttentionStefan K. Nielsen, Laziz U. Abdullaev, Rachel S. Y. Teo, Tan NguyenNeurIPS 2024 · 12 citations
- TMPNN: High-Order Polynomial Regression Based on Taylor Map FactorizationAndrei Ivanov, Stefan Maria AiluroAAAI 2024 · 3 citations
Builds on1
Related papers
- Regression Trees Know CalculusNathan WycoffNeurIPS 2025
- Leveraging Locality and Robustness to Achieve Massively Scalable Gaussian Process RegressionRobert Allison, Anthony Stephenson, Samuel F, Edward O. Pyzer-KnappNeurIPS 2023 · 8 citations
- Meta-NeighborhoodsSiyuan Shan, Yang Li, Junier B. OlivaNeurIPS 2020 · 17 citations
- Revisiting Nearest Neighbor for Tabular Data: A Deep Tabular Baseline Two Decades LaterHan-Jia Ye, Huai-Hong Yin, De-Chuan Zhan, Wei-Lun ChaoICLR 2025
- Extrapolation Towards Imaginary 0-Nearest Neighbour and Its Improved Convergence RateAkifumi Okuno, Hidetoshi ShimodairaNeurIPS 2020 · 2 citations
