Poda de transformers con bandidos conscientes del daño
Un método de poda estructurada post-entrenamiento selecciona unidades funcionales de transformers de lenguaje y visión mediante bandidos multi-brazo, reduciendo la degradación frente a alternativas codiciosas.
Ilustracion contextual ViaMind · no es fotografia del hecho
Lo reportado
Método y resultados
Según arXiv · cs.AI, el artículo 'Damage-Aware Bandit Pruning for Vision and Language Transformers' formula la selección de unidades estructuradas como un problema de bandidos multi-brazo consciente del daño. Se enmascaran temporalmente cabezas de atención y grupos de canales MLP en lotes de calibración, y el daño emparejado reduce la variación entre lotes. En experimentos con GPT-2, OPT, Pythia, Qwen2.5, SmolLM2, ViT-B/16, DeiT-Tiny y Swin-Tiny, los métodos de bandidos suelen reducir la degradación frente a la selección codiciosa presupuestada.
Estado de evidencia: Las afirmaciones materiales incluidas fueron contrastadas con la evidencia enlazada.