Ir para o conteúdo
ModelListadosExterno
Browse files

Rqsa Mil v1

RQSA-MIL (Region-Quality Supervised Attention for Multiple Instance Learning) is a nudity detection model: a MobileNetV2 backbone with a spatial attention gate conditioned on region quality. During training the gate...

TipoOtherVersãoRqsa v1 cleanCnota

Fonte do modelo

Tipo
Other
Tamanho do arquivo
2,66–10,0 MB
Versão
vRqsa v1 clean
Atualizado
22 de out. de 2026
Fonte
Hugging Face

Descrição da fonte

Model card: RQSA-MIL v1

Summary

RQSA-MIL (Region-Quality Supervised Attention for Multiple Instance Learning) is a nudity detection model: a MobileNetV2 backbone with a spatial attention gate conditioned on region quality. During training the gate receives the IoU between each region and ground-truth annotations; at inference it receives a pseudo-IoU of 0.5. Trained with weighted MIL on bag-level labels plus an attention diversity regularizer.

  • Architecture: MobileNetV2 (3.5M params) + IoU-conditioned attention (346K params, ~10% overhead) + linear head
  • Input: 224x224 crops produced by dense sliding-window inference over images normalized to a 690px short side
  • Output: per-region probability; image verdict via per-tier top-3 mean with calibrated dual thresholds (large tier 0.50, small tier 0.78)

Intended use

Defensive content moderation: blurring or flagging NSFW imagery in browsers, user-generated-content platforms, or safety pipelines. Not intended for punitive automated decisions about individuals without human review.

Performance (DS1 test set, dense evaluation)

MetricValue
Recall87.89%
Specificity85.52%
AUC92.98%
TPR at 1% FPR48.65%

Thresholding guidance

  • score = max(large-tier, small-tier) top-3 mean. Use as a UI confidence.
  • score_large is the recommended single-knob threshold (ROC/AUC numbers are computed on it). Lower threshold = stricter blocking, more false positives.
  • The fast preset trades some sensitivity for ~8x fewer patches; production CPU deployments use it for interactive latency.

Limitations

  • Trained primarily on photographic content; drawings, anime, and heavily stylized imagery are out of distribution.
  • Small embedded thumbnails (under ~140px on the normalized image) are scored by the noisier small-window tier only.
  • Skin-tone-heavy but safe content (swimwear, dermatology, classical art) can produce elevated scores; keep a human review path for contested verdicts.
  • The model detects nudity, not consent, age, or context. It must not be used as the sole basis for content takedowns or accusations.

Training data

Trained on a bounding-box-annotated nudity detection dataset (positives) and a nude/non-nude classification dataset (negatives). Datasets are not redistributed with this repository.

Fontes

1 fonte
Hugging FacePrimárioVersão 73dae95ae3b4bf9cb2b6bf41333f51eed5b1abb8
huggingface.co/Shivapreetham17/rqsa-mil-v1/tree/73dae95ae3b4bf9cb2b6bf41333f51eed5b1abb8

Verificado 22 de out.

Download files

Artefatos de modelo

3 artefatos

rqsa_v1_clean.pt

pt · 10,0 MB · SHA-256 fce38c534255…c8a6 · Hugging Face

Download file

rqsa_v1_int8.onnx

onnx · 2,66 MB · SHA-256 fa28d86af0d3…a6b2 · Hugging Face

Download file

rqsa_v1.onnx

onnx · 9,79 MB · SHA-256 78073995159e…ee2a · Hugging Face

Download file

Trechos de fonte

2 trechos

--- license: other tags: - image-classification - content-moderation - not-for-all-audiences --- # Model card: RQSA-MIL v1 ## Summary RQSA-MIL (Region-Quality Supervised Attention for Multiple Instance Learning) is a nudity detection model: a MobileNetV2 backbone with a spatial attention gate conditioned on region quality. During training the gate receives the IoU between each region and ground-truth annotations; at inference it receives a pseudo-IoU of 0.5. Trained with weighted MIL on bag-level labels plus an attention diversity regularizer. - Architecture: MobileNetV2 (3.5M params) + IoU-conditioned attention (346K params, ~10% overhead) + linear head - Input: 224x224 crops produced by dense sliding-window inference over images normalized to a 690px short side - Output: per-region probability; image verdict via per-tier top-3 mean with calibrated dual thresholds (large tier 0.50, small tier 0.78) ## Intended use Defensive content moderation: blurring or flagging NSFW imagery in browsers, user-generated-content platforms, or safety pipelines. Not intended for punitive automated decisions about individuals without human review. ## Performance (DS1 test set, dense evaluation) | Metric | Value | |---|---| | Recall | 87.89% | | Specificity | 85.52% | | AUC | 92.98% | | TPR at 1% FPR | 48.65% | ## Thresholding guidance - `score` = max(large-tier, small-tier) top-3 mean. Use as a UI confidence. - `score_large` is the recommended single-knob threshold (ROC/AUC numbers are computed on it). Lower threshold = stricter blocking, more false positives. - The `fast` preset trades some sensitivity for ~8x fewer patches; production CPU deployments use it for interactive latency. ## Limitations - Trained primarily on photographic content; drawings, anime, and heavily stylized imagery are out of distribution. - Small embedded thumbnails (under ~140px on the normalized image) are scored by the noisier small-window tier only. - Skin-tone-heavy but safe content (swimwear, dermatology, classical art) can produce elevated scores; keep a human review path for contested verdicts. - The model detects nudity, not consent, age, or context. It must not be used as the sole basis for content takedowns or accusations. ## Training data Trained on a bounding-box-annotated nudity detection dataset (positives) and a nude/non-nude classification dataset (negatives)....

Source context: 0 downloads · 0 likes · Pipeline image-classification · Repo Shivapreetham17/rqsa-mil-v1