Skip to main content

Module multitask

Module multitask 

Source
Expand description

§マルチタスク動的マーカー並列スキャンモジュール (1 パス推論)

インジェクション (INJ)、有害性 (TOX)、個人情報漏洩 (PII) の 3 項目を個別推論せず、 系列先頭に [INJ], [TOX], [PII] の動的マーカーを配置して単一のフォワードパスで同時評価する。 これにより推論パス数を 3 回から 1 回に集約し、66% の計算量を削減する。 また、ヘルムホルツ自由エネルギー $F(x) = -T \ln \sum \exp(z_k / T)$ による OOD 異常検知を行う。

Structs§

MultiTaskScanResult
単一パス並列スキャン評価結果。
MultiTaskScannerConfig
マルチタスクスキャナー設定。

Constants§

MARKER_INJECTION
特殊マーカー定義。
MARKER_PII
MARKER_TOXICITY

Functions§

build_multitask_prompt
マルチタスク推論用の合成プロンプト系列を構築する。
compute_free_energy
二値分類マルチタスクにおけるヘルムホルツ自由エネルギーを計算する。
evaluate_multitask_logits
ロジットから各タスクの確率と自由エネルギーを算出し、総合判定を下す。
sigmoid
ロジット列からシグモイド確率を算出する。