Aller au contenu
ModelPrésentéExterne
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...

TypeOtherVersionRqsa v1 cleanCClasse

Source du modèle

Type
Other
Taille du fichier
2,66–10,0 MB
Version
vRqsa v1 clean
Mis à jour
22 oct. 2026
Source
Hugging Face

Description de la source

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.

Sources

1 source
Hugging FacePrincipalVersion 73dae95ae3b4bf9cb2b6bf41333f51eed5b1abb8
huggingface.co/Shivapreetham17/rqsa-mil-v1/tree/73dae95ae3b4bf9cb2b6bf41333f51eed5b1abb8

Vérifié 22 oct.

Download files

Artefacts du modèle

3 artefacts

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

Extraits de sources

2 extraits

--- 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