Confidence is the key: how conformal prediction enhances the generative design of permeable peptides
コンフォーマル予測の校正済み不確実性を RL 報酬に直結する不確実性考慮型生成 ｜ arXiv:2605.05770 [cs.AI] (2026, AstraZeneca)
🎯 予測モデルの生確率でなく CP の校正済み p 値(P0/P1)を RL 報酬に使い、生成を予測器の適用領域内にとどめつつ高信頼で透過と予測される環状ペプチドを効率獲得する。
① 背景と課題

RL 生成(REINVENT/PepINVENT)は予測モデルを報酬に組み込むが、予測器には適用領域があり、RL が訓練分布外の分子を生成すると予測が信頼できなくなる。生の確率は高値を返しうるため「報酬高・不確実性高」空間へ誘導され過信した誤予測を量産。本研究は環状ペプチドの受動膜透過性設計で CP の校正済み不確実性を RL スコアに直接組み込む。

従来は予測モデルの生確率をそのまま報酬とし不確実性の代理とみなすが、校正されておらず適用領域外の信頼性が不明
先行 CP 研究は予測モデル単体の評価にとどまり、生成（RL）ループに組み込んでいない
P1 のみ最大化では Mondrian ICP の P0/P1 同時判断（双対性）を学べない

→ CP を静的指標でなく RL 報酬関数そのものに組み込み、生成（探索）と不確実性定量化を結合する

② 手法: CP-informed RL（PepINVENT + ACP）
CP-informed RL 報酬 PepINVENT (CHUCKLES) 1万非天然AA でマスク補完生成 XGBoost 透過性分類器 ECFP・balanced acc 78%/感度90% Mondrian ICP×10 集約 ACP → P0/P1（クラス別 p 値） soft 報酬 P1>0.2 & P0<0.2→1 / 片方→0.5 / 不成立→0
conformally efficient = (P1>0.2 ∧ P0<0.2)
信頼度ノブ significance=0.2（80%信頼）・REINVENT σ=50・350 epoch

6 報酬関数を比較し soft が最速収束・最多の高信頼透過を生成。基盤 PepINVENT/REINVENT4 は公開。

③ 本研究で示したこと（要点）
  • CP を静的評価でなく RL 報酬関数に組み込んだ初の不確実性考慮型生成（CP × RL 統合）
  • Mondrian ICP の P0/P1 双対性を活かす soft/harsh コンフォーマル効率報酬を設計
  • 生成を予測器の適用領域内にとどめ「過信ヒットの暴走」を回避
  • soft 報酬が約 150/350 epoch で目標到達、最速収束かつ多様性も最高
④ 主な結果 (a) 過信の可視化
0.51 0.15 開始 0.87 0.23 350ep RLスコア(生) P1(校正) 生報酬の見かけ vs 校正

生スコアは 0.87 まで上がるが P1 は 0.23 止まり・P0 も 0.2 を割らない＝過信のボトルネックは P0。

④ 主な結果 (b) 収束速度
150 soft 報酬 350 生報酬 目標到達 epoch (↓良)

soft は約 150/350 epoch で目標（有効分子50%が高信頼透過）に到達、生報酬は未到達。

④ 主な結果 (c) CP 妥当性
82 非透過(Class0) 85 透過(Class1) CP クラス妥当性 (%)

指定信頼度 80% で妥当性を担保。生 XGBoost は balanced acc 78%/感度 90%。

④ 主な結果 (d) 分類器・要点
balanced acc 78% / 感度 90%
生 XGBoost 透過性分類器（CycPeptMPDB PAMPA, 6876 点）
soft 報酬が最良
最速収束＋最多の高信頼透過ペプチド＋最大多様性
適用領域ゲート
None(領域外)/Both(判別不能) を明示し過信を抑制
⑤ テイクホームメッセージ
🎯 校正報酬
生確率でなく CP の P0/P1 を RL 報酬に直結し過信を抑制。
🛡️ 領域内維持
生成を予測器の適用領域内にとどめ暴走を回避。
⚡ 最速収束
soft 報酬が約 150/350 epoch で目標到達、多様性も最高。
🔁 双対性
P1 のみでは不十分、P0/P1 の同時判断が鍵。
報酬関数の比較
報酬定式結果
生 P(class1)校正なし確率RLスコア↑だが過信(P0>0.2)
P1 のみP1 最大化双対性を学べず限定的
harshP1≥0.2 & P0≤0.2→1学習可・安定
soft（本研究）片方成立で 0.5 の3段最速収束・最多・最多様
本研究のインパクト
  • lib/molgen: 既存スコアラーを ACP(10 ICP)でラップし significance_level(既定0.2)を信頼度ノブとして公開、soft/harsh 報酬トランスフォームを選択可能に。RL/REINVENT 型生成で適用領域外の過信ヒットを抑え収束を速め信頼性を底上げ
  • lib/docking: LBVS/SBVS の QSAR スコアフィルタに CP ラッパを付け、点スコアを盲信せず None(領域外)/Both(判別不能) を弾く applicability-domain ゲートに
  • lib/fep: MM-GBSA/ML 親和性サロゲートを前段フィルタに使う際 ACP で validity/efficiency を付与し、適用領域内の化合物だけ高コスト FEP に回し計算資源を節約
Confidence is the key: how conformal prediction enhances the generative design of permeable peptides. arXiv:2605.05770 [cs.AI]. 2026 (AstraZeneca). https://arxiv.org/abs/2605.05770