Skip to content
Β 
Β 

Latest commit

Β 

History

20 Commits

Folders and files

NameName
Last commit message
Last commit date
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🌿 ViRel-Budget

Reliability-constrained visual-token budgeting for green vision-language inference
Accepted for an oral presentation at the ACM MM 2026 GreenMM Workshop

Cases Models Pruners Robustness Energy Status Tests License DOI: 10.5281/zenodo.22014989

ViRel-Budget asks a concrete systems question: how aggressively can a vision-language model prune visual tokens while preserving its dense answer and intervention-defined response behavior? It turns development-only supervision into a prespecified controller, executes one selected action per query, and evaluates the complete path using accuracy, unsafe acceptance, GPU energy, latency, memory, carbon estimates, and reliability-adjusted efficiency.

GitHub repository: https://github.com/StableTradeAtlas/ViRel-Budget

Hugging Face dataset: https://huggingface.co/datasets/StableTradeAtlas/ViRel-Budget

Important

ViRel-Budget measures intervention-defined visual reliance; it does not claim to certify universal semantic grounding. Intervention outputs supervise and evaluate the controller but are never available to it at deployment.

✨ Release snapshot

Dimension Public artifact
Evaluation 1,200-query development pool + 900 sealed prospective queries
Backends LLaVA-1.5-7B and LLaVA-1.5-13B; dense SmolVLM-500M scale audit
Pruning FastV, SCOPE, and deterministic Random control
Execution Prespecified metadata controller; one backend invocation per query
Robustness Three group-disjoint, stratified 210-case workload draws
Hardware One NVIDIA RTX PRO 6000 Blackwell Server Edition
Environmental accounting NVIDIA device telemetry, prospective CodeCarbon, disclosed PUE and grid-intensity scenarios

πŸ“Š Main finding

Across the separate 630-query repeated-workload population, SCOPE was the only pruner with positive energy reduction on every draw:

Backend System GPU energy vs. dense 95% CI Unsafe acceptance Dense avoidance
7B SCOPE 11.87% lower [7.19%, 16.30%] 0.63% 33.97%
13B SCOPE 10.62% lower [7.48%, 13.46%] 1.27% 50.00%
7B FastV 3.11% higher [βˆ’6.84%, 0.75%] reduction 0.79% 61.27%
13B FastV 1.35% higher [βˆ’4.13%, 1.39%] reduction 2.54% 77.78%
7B Random 5.96% higher [βˆ’11.40%, βˆ’0.95%] reduction 2.54% 55.56%
13B Random 3.92% higher [βˆ’7.40%, βˆ’0.55%] reduction 1.43% 52.22%

The result separates token efficiency from realized device-energy efficiency: avoiding dense inference or retaining fewer tokens does not automatically reduce measured energy. The unsafe rates above belong to the separate 630-query repeated-workload population; prospective reliability results are reported independently under results/prospective.

See results/replication/summary and docs/results.md.

🧠 Method in one view

  1. Construct strict labels that separately record task correctness, dense-answer agreement, intervention-response preservation, and their conjunction.
  2. Learn a low-capacity controller from the 1,200-case development pool using deployment-available metadata only.
  3. Prespecify the controller, risk threshold, action space, fallback, manifests, and measurement boundary.
  4. Execute exactly one selected action on each of 900 group-isolated prospective cases.
  5. Reveal hidden evaluation labels only after execution is complete.
  6. Repeat the principal systems comparison on three independently sampled, group-disjoint 210-case workloads.

Full details: docs/methodology.md, docs/datasets.md, and docs/measurement.md.

πŸ—‚οΈ Repository map

ViRel-Budget/
β”œβ”€β”€ virel_budget/          # installable framework and backend adapters
β”œβ”€β”€ scripts/               # data, execution, measurement, analysis, and audits
β”œβ”€β”€ configs/               # recorded development, prospective, and repeated-workload protocols
β”œβ”€β”€ data/                  # manifests and lightweight test fixtures (not full images)
β”œβ”€β”€ results/               # publication-facing summaries and compact raw repeated-workload data
β”‚   └── feature_analysis/  # development-only grouped feature analysis
β”œβ”€β”€ patches/               # deterministic Random pruning patch for official FastV
β”œβ”€β”€ tests/                 # unit and restart/integrity tests
└── docs/                  # methodology, artifacts, measurement, limitations

Large historical checkpoints, model weights, source images, telemetry streams, exploratory plots, paper source, handoffs, and obsolete experiments are intentionally excluded from the public release.

πŸš€ Quick start

python -m venv .venv
source .venv/bin/activate
python -m pip install -e .
python -m unittest discover -s tests -v
virel-budget --help

Model-backed reproduction additionally needs the official FastV/SCOPE environments and model weights; see REPRODUCIBILITY.md. Saved-result inspection does not require a GPU.

πŸ”Ž What drives the controller?

A development-only grouped analysis compares question structure, task/source indicators, and semantic-keyword indicators across all six controllers. Task/source indicators have the largest mean absolute standardized coefficient in all 24 budget-specific regressions and are the strongest single feature family for dense avoidance in four of six controllers. Removing them reduces dense avoidance in five controllers by 18.33–32.08 percentage points; 13B FastV is the exception.

This result shows substantial reliance on task identity and does not establish transfer to unseen task families. No image-derived metadata are used. The complete verified outputs are under results/feature_analysis.

♻️ Environmental accounting

The recoverable experimental program contains 66 runs, 19.17 GPU-hours, and 4.8198 measured GPU kWh. Under the disclosed central assumptions (Iceland region, 0.03283 kgCOβ‚‚e/kWh, PUE 1.2), this corresponds to 0.1899 kgCOβ‚‚e; the scenario range is 0.0795–0.5398 kgCOβ‚‚e. These are operational estimates, not lifecycle carbon measurements. CodeCarbon totals are reported separately because host-wide CPU/RAM estimates do not match the pod allocation.

See results/carbon for the run ledger and docs/measurement.md for system boundaries.

⚠️ Scope and limitations

  • The controller is model/pruner specific and tested on the stated workloads and hardware.
  • SCOPE shows consistent device-energy savings across all three workload draws; FastV does not on this implementation.
  • Carbon is estimated from measured GPU energy plus disclosed regional assumptions.
  • The dense SmolVLM-500M run is a model-scale audit, not a like-for-like alternative pruner evaluation.
  • Intervention-defined safety measures behavioral preservation, not whether a model uses the correct visual evidence.

See docs/limitations.md before reusing the claims.

πŸ“œ Citation and license

Citation metadata is in CITATION.cff. Code is released under the Apache License 2.0. Source images and external model/pruning implementations retain their original licenses and are not redistributed here. Dataset manifests are provided for research reproducibility and remain subject to their source datasets' terms.

Open data and code statement

To support transparency, reproducibility, and reuse, we release evaluation manifests, derived annotations, configuration files, and accompanying documentation on Hugging Face. This repository releases the source code, evaluation scripts, summarized experimental outputs, and compact raw repeated-workload records. Source images, model weights, and the complete historical intermediate grids are not redistributed; users must obtain external materials from their original providers under the applicable licenses and terms.

πŸ™ Acknowledgments

Sean Wan and Shilin Ou gratefully acknowledge support from the Summer Research Scholars Program at Duke Kunshan University (DKU) and from the DKU Library grant supporting the Shanghai Library Open Data Contest project entitled "StableTradeAtlas", both under the supervision of Prof. Luyao Zhang.