cs.CVAug 24, 2026

How Merge-Tolerant Are Vision Transformers for Wheat Phenotyping?

Authors: Simon Ravé, Pejman Rasti, David Rousseau

Organizations: LARIS, University of Angers, Angers, France · UMR INRAE-IRHS, Angers, France

Abstract

Vision-based wheat phenotyping requires repeated measurements under deployment constraints, from growth-stage recognition to wheat-head counting and organ segmentation. Plain Vision Transformers (ViTs) provide a common architecture for these tasks, but quadratic attention limits high-throughput and edge inference. Training-free token merging is attractive because it can be inserted into trained models without retraining. We provide a systematic benchmark of ToMe and Mutual Pair Merging across growth-stage classification, wheat-head detection, and wheat-organ segmentation, measuring task quality, throughput, token count, and peak GPU memory, with additional Raspberry Pi 5 measurements. The benchmark reveals a clear hierarchy: classification is highly merge-tolerant, while detection and segmentation are constrained by repeated instances, thin organs, dense boundaries, reconstruction, and runtime overhead. Optimized attention backends can erase apparent speedups, so deployment value must be profiled on the target runtime rather than inferred from token count.

Explore similar work

CardsList
  1. Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TM

    Sep 16, 2026Victor Bercy, Martyna Poreba, Michal Szczepanski +1Visual Token MergingEfficient ViTs

  2. MPM: Mutual Pair Merging for Efficient Vision Transformers

    Apr 7, 2026Simon Ravé, Pejman Rasti, David RousseauVisual Token MergingEfficient ViTs

  3. MergeOver: Post-Training Token Merging for Recursive Vision Transformers

    Aug 13, 2026Junseo Kim, Uraz Odyurt, Amirreza YousefzadehVision TransformerEfficient ViTs