A Novel Approach to Evaluating Evaluation Metrics for Multi-Output Structured Prediction
Organizations: The University of Texas at Dallas
Abstract
In multi-output structured prediction tasks, while only one ground truth label may be provided in the training data, multiple equally valid outputs may be possible, making reliable evaluation a persistent challenge. We postulate that human evaluators implicitly use task-specific invariants, e.g., object boundaries in colorized images or named entities in translations, to judge if an output is acceptable. Under this assumption, we introduce a notion of approximate task-specific invariants and use them as diagnostic tools to evaluate a variety of existing metrics for vision and language tasks. We use these task invariants as part of a framework to systematically test metric reliability by encouraging domain-relevant invariants in model outputs via an augmented loss function. In our experiments, we observe that enforcing invariants with an augmented loss yields substantial improvements in popular distributional metrics while more traditional metrics change only marginally. Through this invariants-driven evaluation, we expose where standard metrics fail to detect meaningful differences, and we highlight the conditions under which distributional metrics succeed or still fall short.
Figures & tables
| Main Task | Invariant Task(s) |
|---|---|
| Image Colorization | Image Segmentation, Depth Estimation, Edge Detection |
| Machine Translation (NMT) | Named Entity Recognition (NER), Sentiment Analysis, Paraphrase Detection |
| Image Captioning | Semantic Role Labeling (SRL), Scene Graph Consistency, Named Entity Recognition (NER) |
| Text Summarization | Named Entity Recognition (NER), Sentiment Analysis, Coreference Resolution |
| Video Prediction | Optical Flow Estimation, Object Tracking, Action Recognition |
| Natural Language Inference | Paraphrase detection, Sentiment Analysis |
| Task | Image Colorization | German to English Neural Machine Translation | Image Captioning |
| Base Networks | ChromaGAN ( Vitoria, Raad, and Ballester 2020 ) , ColorFormer ( Ji et al. 2022 ) , and DDColor ( Kang et al. 2023 ) | BiBERT ( Xu, Van Durme, and Murray 2021 ) and Bi-SumCut ( Gao et al. 2022 ) | SmallCap ( Ramos et al. 2023 ) and ClipCap ( Mokady, Hertz, and Bermano 2021 ) |
| Loss Function ( ) | Pixel‐reconstruction MSE, L1-loss, Perceptual-loss, GAN-loss, Colorfulness-loss | Label Smoothed Cross Entropy: Cross-entropy loss measuring predicted vs target word probability differences with label smoothing for better generalization | Cross Entropy Loss: Token-level classification loss between predicted and ground truth caption sequences |
| Invariant Networks | Depth: MiDaS ( Ranftl et al. 2020 ) Segmentation: Side Adapter Network (SAN) ( Xu et al. 2023 ) | NER: Hugging Face’s ( Wolf et al. 2019 ) Sentiment: Hugging Face’s distilbert-base- uncased-finetuned-sst-2-english | NER: Hugging Face’s ( Wolf et al. 2019 ) Scenegraph: sngparser ( Wu et al. 2019 ) ( Schuster et al. 2015 ) |
| Invariant Loss ( ) | SegmentationSANLoss: Dice loss between predicted vs GT SAN segmentation masks prevents color bleeding DepthMidasLoss: SILog loss between predicted vs GT MiDaS depth maps preserves spatial structure | NER Loss: KL divergence between predicted vs GT BERT-NER token classifications for entity preservation and Sentiment Loss: KL divergence between predicted vs GT DistilBERT sentiment logits | NER Loss: KL divergence between predicted vs GT BERT-NER token classifications for entity preservation. Scene Graph Loss: Symmetric difference ratio of entities and relations between predicted vs GT scene graphs |
| Method | #P | ImageNet (val50k) | COCO-Stuff(val5k) | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| FDD | FID | CMMD | LPIPS | CF | PSNR | FDD | FID | CMMD | LPIPS | CF | PSNR | ||
| ChromaGAN | 173.9M | 42.36 | 5.81 | 0.597 | 0.236 | 27.41 | 23.65 | 94.90 | 16.73 | 0.715 | 0.233 | 27.89 | 23.66 |
| +Seg | 173.9M | 29.38 | 5.44 | 0.679 | 0.236 | 29.36 | 23.76 | 81.07 | 15.50 | 0.603 | 0.233 | 29.56 | 23.82 |
| +Depth | 173.9M | 28.55 | 5.27 | 0.565 | 0.231 | 28.73 | 23.83 | 79.28 | 15.29 | 0.659 | 0.229 | 28.95 | 23.91 |
| +Seg+Depth | 173.9M | 28.86 | 5.06 | 0.599 | 0.231 | 28.69 | 23.83 | 80.00 | 14.58 | 0.597 | 0.229 | 29.99 | 23.82 |
| ColorFormer | 44.8M | 21.16 | 2.00 | 0.318 | 0.137 | 27.57 | 24.91 | 59.38 | 9.03 | 0.377 | 0.134 | 28.30 | 24.70 |
| Model | #P | COCO-Stuff (val5k) | Flickr30k (val1k) | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| FBD-4 | FBD | BERTScore | CIDEr | BLEU-4 | METEOR | FBD-4 | FBD | BERTScore | CIDEr | BLEU-4 | METEOR | ||
| SmallCap | 7M | 62.60 | 41.90 | 0.9224 | 118.49 | 36.68 | 27.90 | 58.45 | 39.12 | 0.9189 | 112.30 | 34.82 | 26.45 |
| +NER | 7M | 58.82 | 41.89 | 0.9223 | 116.29 | 36.07 | 27.91 | 55.20 | 39.08 | 0.9187 | 110.15 | 34.25 | 26.48 |
| +SceneGraph | 7M | 54.12 | 41.90 | 0.9218 | 115.47 | 35.68 | 27.76 | 51.85 | 39.15 | 0.9182 | 109.42 | 33.89 | 26.32 |
| +NER+SceneGraph | 7M | 56.52 | 41.89 | 0.9226 | 117.48 | 36.52 | 27.95 | 53.78 | 39.11 | 0.9191 | 111.68 | 34.71 | 26.51 |
| ClipCap | 43M | 48.20 | 38.45 | 0.9301 | 125.80 | 39.15 | 29.20 | 44.60 | 36.25 | 0.9267 | 119.45 | 37.28 | 27.85 |
| Model | #P | IWSLT-14 | WMT-14 | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| FBD | FBD-4 | BERTScore | BLEU-4 | METEOR | COMET | FBD | FBD-4 | BERTScore | BLEU-4 | METEOR | COMET | ||
| Bi-SimCut | 65M | 9.87 | 102.15 | 0.8871 | 37.83 | 0.6412 | 0.5743 | 15.24 | 135.42 | 0.8952 | 34.47 | 0.5836 | 0.6124 |
| +NER | 65M | 8.42 | 101.89 | 0.8867 | 36.95 | 0.6358 | 0.5739 | 14.18 | 134.76 | 0.8948 | 33.82 | 0.5794 | 0.6119 |
| +SeAM | 65M | 8.51 | 103.14 | 0.8868 | 37.02 | 0.6361 | 0.5737 | 14.26 | 136.23 | 0.8949 | 33.89 | 0.5798 | 0.6117 |
| +NER+SeAM | 65M | 8.38 | 102.67 | 0.8869 | 37.18 | 0.6373 | 0.5741 | 14.09 | 135.58 | 0.8950 | 34.06 | 0.5809 | 0.6121 |
| BiBERT | 110M | 8.99 | 100.24 | 0.8930 | 38.20 | 0.6597 | 0.5899 | 13.76 | 133.18 | 0.9012 | 35.24 | 0.6023 | 0.6287 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Example 1 | |
|---|---|
| Source Sentence: | “heinrich , auf der anderen seite , wird zum geburtstermin geboren , er ist also am 1 marz 1980 geboren ,” |
| Target Sentence: | “henry , by contrast , is born at term , so he ’ s born on march 1 , 1980 .” |
| Target Logits: | 0.99992 0.00001 0.00000 0.00003 0.00001 0.00001 0.00001 0.00001 0.00001 O B-MIS I-MIS B-PER I-PER B-ORG I-ORG B-LOC I-LOC |
| BiBERT+NER Pred Sentence: | “henry , on the other hand , is born at the time of birth , so he was born on march 1 , 1980 .” |
| BiBERT+NER Logits: | 0.99993 0.00001 0.00000 0.00002 0.00001 0.00001 0.00001 0.00001 0.00001 O B-MIS I-MIS B-PER I-PER B-ORG I-ORG B-LOC I-LOC |
| BiBERT Pred Sentence: | “and tom , on the other hand , is born at the time of birth , so he was born on march 1 , 1980 .” |
| Model | Dataset | Best | Batch Size | Epochs / Iterations | Hyperparameters Tried ( ) |
|---|---|---|---|---|---|
| ChromaGAN | ImageNet | - | 10 | 5 epochs | - |
| ChromaGAN + Seg | ImageNet | 0.02 | 10 | 15 epochs | 0.0002, 0.002, 0.02, 0.05, 0.2 |
| ChromaGAN + Depth | ImageNet | 0.0001 | 10 | 15 epochs | 0.0001, 0.001, 0.005, 0.01 |
| ChromaGAN + Seg + Depth | ImageNet | 0.02,0.0001 | 10 | 15 epochs | (0.02,0.0001),(0.02,0.005),(0.01,0.001) |
| ColorFormer | ImageNet | - | 16 | 200,000 iterations | - |
| ColorFormer + Seg | ImageNet | 0.05 | 16 | 600,000 iterations | 0.0001, 0.001, 0.01, 0.05, 0.1, 0.2 |
| Model | Dataset | Best | Batch Size | Epochs | Hyperparameters Tried ( ) |
|---|---|---|---|---|---|
| BiBERT | IWSLT-14 | - | 4096 tokens | 75 | - |
| BiBERT + NER | IWSLT-14 | 100 | 4096 tokens | 75 | 10, 50, 100, 500, 1000 |
| BiBERT + SeAM | IWSLT-14 | 50 | 4096 tokens | 75 | 10, 50, 100, 500, 1000 |
| BiBERT + NER + SeAM | IWSLT-14 | 100, 50 | 4096 tokens | 75 | (100,50), (50,50), (100,100) |
| Bi-SimCut | IWSLT-14 | - | 4096 tokens | 75 | - |
| Bi-SimCut + NER | IWSLT-14 | 200 | 4096 tokens | 75 | 50, 100, 200, 500, 1000 |
| Model | Dataset | Best | Batch Size | Epochs | Hyperparameters Tried ( ) |
|---|---|---|---|---|---|
| SmallCap | COCO | - | 32 | 10 | - |
| SmallCap + NER | COCO | 0.5 | 32 | 10 | 0.1, 0.3, 0.5, 0.7, 1.0 |
| SmallCap + SceneGraph | COCO | 1.0 | 32 | 10 | 0.5, 1.0, 2.0, 5.0 |
| SmallCap + NER + SG | COCO | 0.5, 1.0 | 32 | 10 | (0.5,1.0), (0.3,1.0), (0.5,2.0) |
| ClipCap | COCO | - | 64 | 10 | - |
| ClipCap + NER | COCO | 0.3 | 64 | 10 | 0.1, 0.3, 0.5, 0.7, 1.0 |
Automated Evaluation
Automated evaluation develops metrics, benchmarks, and AI judges to assess model outputs and system behavior at scale.
See all 398 papers