← MIC coverage of Transformers model types

docling-project/docling-layout-heron

huggingface-mic/docling-layout-heron-mic@a4c39589d12b ← docling-project/docling-layout-heron@8f39ad3c0b4c · RTDetrV2ForObjectDetection · random weights

CI: no failure on main in 30 days · MIC failures dashboard

Original42.8Mparameters
Fixture10.9Mparameters
Layer stacks11compared
Findings0to review

✓ Same architecture, smaller

Every module class, layer kind and checked property of the original is in the fixture.

Layer stacks

One square per layer, colored by kind: the layer's module classes and type attributes. Hover a square for its index and kind.

model.backbone.model.encoder.stages

Original · 4
Fixture · 4
KindOriginalFixture
no AvgPool2d11✓ kept
AvgPool2d33✓ kept

model.encoder_input_proj

Original · 3
Fixture · 3
KindOriginalFixture
Conv2d33✓ kept

model.encoder.aifi

Original · 1
Fixture · 1
KindOriginalFixture
AIFILayer11✓ kept

model.encoder.lateral_convs

Original · 2
Fixture · 2
KindOriginalFixture
ConvNormLayer22✓ kept

model.encoder.fpn_blocks

Original · 2
Fixture · 2
KindOriginalFixture
CSPRepLayer22✓ kept

model.encoder.downsample_convs

Original · 2
Fixture · 2
KindOriginalFixture
ConvNormLayer22✓ kept

model.encoder.pan_blocks

Original · 2
Fixture · 2
KindOriginalFixture
CSPRepLayer22✓ kept

model.decoder_input_proj

Original · 3
Fixture · 3
KindOriginalFixture
Conv2d33✓ kept

model.decoder.layers

Original · 6
Fixture · 2
KindOriginalFixture
DecoderLayer62✓ kept

model.decoder.bbox_embed

Original · 6
Fixture · 2
KindOriginalFixture
Linear62✓ kept

bbox_embed

Original · 6
Fixture · 2
KindOriginalFixture
Linear62✓ kept

Where the parameters are

Share of each model's parameters per component. Vocabulary and patch sizes are kept, so embeddings usually take a larger share of the fixture.

Original
Fixture
ComponentOriginalShareFixtureShare
model.backbone.model.encoder.stages23.4M54.8%32.4K0.3%
model.decoder.layers6.0M14.0%102.6K0.9%
model.encoder.pan_blocks4.5M10.4%4.5M41.0%
model.encoder.fpn_blocks4.5M10.4%4.5M41.0%
model.encoder.downsample_convs1.2M2.8%1.2M10.8%
model.encoder_input_proj919.0K2.1%58.9K0.5%
model.decoder.bbox_embed795.7K1.9%17.2K0.2%
9 others1.5M3.5%566.5K5.2%

Preserved properties

ConfigPropertyOriginalFixture
modelmodel_typert_detr_v2rt_detr_v2✓ kept
modeltie_word_embeddingsTrueTrue✓ kept
backbone_configmodel_typert_detr_resnetrt_detr_resnet✓ kept
backbone_confighidden_actrelurelu✓ kept
backbone_confignum_channels33✓ kept

Module classes

Instances of each class in the built model (prefix RTDetrV2 omitted).

ClassOriginalFixture
Linear12149
Conv2d9761
ReLU5717
Identity5632
FrozenBatchNorm2d5519
RTDetrResNetConvLayer5115
BatchNorm2d4242
ConvNormLayer3636
Sequential3523
ModuleList2820
SiLUActivation2424
LayerNorm219
RTDetrResNetBottleNeckLayer164
MLPPredictionHead146
RepVggBlock1212
MLP73
SelfAttention73
DecoderLayer62
MultiscaleDeformableAttention62
RTDetrResNetShortCut44
RTDetrResNetStage44
CSPRepLayer44
AvgPool2d33
Embedding11
GELUActivation11
MaxPool2d11
RTDetrResNetBackbone11
RTDetrResNetEmbeddings11
RTDetrResNetEncoder11
AIFILayer11
ConvEncoder11
Decoder11
EncoderLayer11
ForObjectDetection11
HybridEncoder11
Model11
SinePositionEmbedding11

Config changes

9 fields changed, 74 unchanged.

FieldOriginalFixture
backbone_config.depths[3, 4, 6, 3][1, 1, 1, 1]
backbone_config.hidden_sizes[256, 512, 1024, 2048][16, 32, 64, 128]
d_model25664
decoder_attention_heads84
decoder_ffn_dim1024128
decoder_layers62
encoder_attention_heads84
encoder_ffn_dim1024128
encoder_in_channels[512, 1024, 2048][32, 64, 128]