Rotation-Robust Regression with Convolutional Model Trees

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Hongyi, Armstrong, William Ward, Xu, Jun
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915716533321728
author Li, Hongyi
Armstrong, William Ward
Xu, Jun
author_facet Li, Hongyi
Armstrong, William Ward
Xu, Jun
contents We study rotation-robust learning for image inputs using Convolutional Model Trees (CMTs) [1], whose split and leaf coefficients can be structured on the image grid and transformed geometrically at deployment time. In a controlled MNIST setting with a rotation-invariant regression target, we introduce three geometry-aware inductive biases for split directions -- convolutional smoothing, a tilt dominance constraint, and importance-based pruning -- and quantify their impact on robustness under in-plane rotations. We further evaluate a deployment-time orientation search that selects a discrete rotation maximizing a forest-level confidence proxy without updating model parameters. Orientation search improves robustness under severe rotations but can be harmful near the canonical orientation when confidence is misaligned with correctness. Finally, we observe consistent trends on MNIST digit recognition implemented as one-vs-rest regression, highlighting both the promise and limitations of confidence-based orientation selection for model-tree ensembles.
format Preprint
id arxiv_https___arxiv_org_abs_2601_04899
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Rotation-Robust Regression with Convolutional Model Trees
Li, Hongyi
Armstrong, William Ward
Xu, Jun
Computer Vision and Pattern Recognition
Machine Learning
We study rotation-robust learning for image inputs using Convolutional Model Trees (CMTs) [1], whose split and leaf coefficients can be structured on the image grid and transformed geometrically at deployment time. In a controlled MNIST setting with a rotation-invariant regression target, we introduce three geometry-aware inductive biases for split directions -- convolutional smoothing, a tilt dominance constraint, and importance-based pruning -- and quantify their impact on robustness under in-plane rotations. We further evaluate a deployment-time orientation search that selects a discrete rotation maximizing a forest-level confidence proxy without updating model parameters. Orientation search improves robustness under severe rotations but can be harmful near the canonical orientation when confidence is misaligned with correctness. Finally, we observe consistent trends on MNIST digit recognition implemented as one-vs-rest regression, highlighting both the promise and limitations of confidence-based orientation selection for model-tree ensembles.
title Rotation-Robust Regression with Convolutional Model Trees
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2601.04899