Kwon & Ham's arXiv paper enables AI knowledge transfer across feature sets
arXiv:2608.12403 shows how to reuse black-box models on entirely new inputs
A new paper from researchers Oh-Ran Kwon and Daeyoung Ham tackles a fundamental problem in machine learning: how to reuse a pre-trained black-box model when your data has different features than what the model expects. The method, detailed in arXiv:2608.12403, decomposes the target regression function into a transferable component—knowledge the black box can inform—and a non-transferable component unique to the new feature space. The authors propose a two-step neural network procedure: first estimate the transferable part using abundant unlabeled feature pairs that bridge the two input spaces, then estimate the non-transferable part using limited labeled data. This decomposition allows the model to leverage pre-trained knowledge without needing direct access to the black box's internals or requiring the input spaces to match.
The theoretical contributions are significant. The authors derive prediction risk bounds that strictly improve on non-transfer alternatives when the non-transferable component is small or smooth, and importantly, the procedure adapts to either case automatically. Under additional conditions, the worst-case risk achieves a strictly smaller polynomial order than the minimax risk from labeled data alone. The framework also extends to multiple black boxes, each operating on its own input space, and shows that aggregating them can reduce prediction error relative to the best single black box. Simulated and real-world experiments demonstrate practical value, making this a promising approach for scenarios like domain adaptation, multi-source learning, and leveraging proprietary APIs. For professionals, this means pre-trained models—even opaque ones—can be adapted to custom data schemas, potentially saving massive retraining costs.
- Two-step neural network separates transferable vs. non-transferable knowledge from a black-box predictor
- Uses abundant unlabeled feature pairs plus limited labels to bridge distinct input spaces
- Achieves strictly lower worst-case prediction risk than minimax estimation with labeled data alone
Why It Matters
Enables reusing any pre-trained black-box model on new feature spaces, cutting retraining costs and unlocking proprietary APIs for custom datasets.