연구진은 디퓨전 트랜스포머(DiT) 추론 속도를 높이는 SoftCap 기술을 제안했어요. SoftCap은 중간 특징을 캐싱하는 방식에, 경로 예측 관찰기와 소프트 예산 PI 컨트롤러를 결합합니다.
경로 예측 관찰기는 캐시 위험을 추정하고, 소프트 예산 PI 컨트롤러는 연산량에 따라 Full 단계 실행 기준을 조절해 이미지 품질과 연산량 균형을 맞춥니다.
FLUX.1-dev 환경에서 SoftCap은 SpeCa와 유사한 연산량에서 ImageReward를 0.967에서 0.981로, LPIPS-Full을 0.518에서 0.498로 향상시켰어요.