ShreyashDhoot/KTO_trial

---
language: en
license: mit
task_categories:
- image-classification
tags:
- image
- safety
- adversarial
- inpainting
- kto
dataset_info:
features:
- name: "image"
dtype: image
- name: "prompt"
dtype: string
- name: "id"
dtype: string
- name: "disturbing"
dtype: int8
- name: "hate"
dtype: int8
- name: "illegal activity" # Keep the quotes here
dtype: int8
- name: "safe"
dtype: int8
- name: "sexual"
dtype: int8
- name: "violence"
dtype: int8
- name: "feathered_mask"
dtype: image
- name: "image_masked_removed"
dtype: image
---

# KTO Training Dataset

Processed from [kricko/cleaned_auditor](https://huggingface.co/datasets/kricko/cleaned_auditor) using the Auditor model.

## Description
Each example contains the original image alongside adversarial heatmaps, feathered masks,
and masked images with detected unsafe regions blacked out.

## Features
| Column | Type | Description |
|---|---|---|
| `image` | Image | Original input image |
| `prompt` | string | Text prompt associated with the image |
| `id` | string | Unique identifier |
| `disturbing` | int8 | Disturbing content score |
| `hate` | int8 | Hate content score |
| `illegal activity` | int8 | Illegal activity score |
| `safe` | int8 | Safe content score |
| `sexual` | int8 | Sexual content score |
| `violence` | int8 | Violence content score |
| `feathered_mask` | Image | Feathered adversarial mask (sigma=5, 75th percentile threshold) |
| `image_masked_removed` | Image | Original image with adversarial regions blacked out |

## Processing Details
- **Threshold**: 75th percentile of adversarial heatmap
- **Morphological dilation**: 5×5 kernel, 2 iterations
- **Feathering**: Gaussian blur sigma=5.0
- **Total examples**: 27553

语言：英语
许可证：MIT协议
任务类别：
- 图像分类
标签：
- 图像
- 安全
- 对抗性（adversarial）
- 图像修复（inpainting）
- KTO
数据集信息：
特征：
- 字段名：`image`，数据类型：图像
- 字段名：`prompt`，数据类型：字符串
- 字段名：`id`，数据类型：字符串
- 字段名：`disturbing`，数据类型：int8
- 字段名：`hate`，数据类型：int8
- 字段名：`illegal activity`，数据类型：int8
- 字段名：`safe`，数据类型：int8
- 字段名：`sexual`，数据类型：int8
- 字段名：`violence`，数据类型：int8
- 字段名：`feathered_mask`，数据类型：图像
- 字段名：`image_masked_removed`，数据类型：图像

# KTO训练数据集

本数据集基于Auditor模型，从[kricko/cleaned_auditor](https://huggingface.co/datasets/kricko/cleaned_auditor)数据集处理得到。

## 数据集说明
每个样本均包含原始输入图像，以及对抗性热力图、羽化掩码与经检测的不安全区域涂黑后的掩码图像。

## 特征说明
| 字段名 | 数据类型 | 描述 |
|---|---|---|
| `image` | 图像 | 原始输入图像 |
| `prompt` | 字符串 | 与该图像关联的文本提示 |
| `id` | 字符串 | 唯一标识符 |
| `disturbing` | int8 | 令人不安内容评分 |
| `hate` | int8 | 仇恨内容评分 |
| `illegal activity` | int8 | 非法活动评分 |
| `safe` | int8 | 安全内容评分 |
| `sexual` | int8 | 色情内容评分 |
| `violence` | int8 | 暴力内容评分 |
| `feathered_mask` | 图像 | 羽化对抗掩码（sigma=5，75%分位数阈值） |
| `image_masked_removed` | 图像 | 对抗区域被涂黑的原始图像 |

## 处理细节
- **阈值**：对抗热力图的75%分位数
- **形态学膨胀**：5×5卷积核，迭代2次
- **羽化处理**：高斯模糊sigma=5.0
- **总样本数**：27553