of retrieved passages prepended to input, or pre-train a retriever and LM jointly, followed by few-shot fine-tuning on task datasets (Izacard et al., 2022b). While prior work often retrieves only once at the beginning, Jiang et al. (2023) propose to adaptively retrieve passages for generation on top of a proprietary LLM or Schick et al. (2023) train an LM to generate API calls for named entities. Yet, the improved task performance of such approaches often comes at the expense of runtime efficiency (Mallen et al., 2023), robustness to irrelevant context (Shi et al., 2023), and lack of attributions (Liu et al., 2023; Gao et al., 2023). We introduce a method to train an arbitrary LM to learn to use retrieval _on-demand_ for diverse instruction-following queries and introduce controlled generation guided by reflections tokens to further improve generation quality and attributions.
입력에 앞에 붙는 검색된 구절을 사전 검색하거나, retriever와 LM을 공동으로 pre-train한 다음 작업 데이터 세트에 대한 few-shot Fine-tuning(Izacard et al., 2022b)을 수행합니다. 이전 작업에서는 처음에 한 번만 검색하는 경우가 많았지만, Jiang et al. (2023)은 독점 LLM 위에 생성을 위해 적응적으로 구절을 검색하거나 Schick et al. (2023)은 명명된 엔티티에 대한 API 호출을 생성하는 LM을 훈련할 것을 제안합니다. 그러나 이러한 접근 방식의 향상된 작업 성능은 종종 runtime efficiency(Mallen et al., 2023), 관련 없는 context에 대한 robustness(Shi et al., 2023), 기여도 부족(Liu et al., 2023; Gao et al., 2023)의 희생으로 이어집니다. 다양한 Instruction에 따라 검색을 _on-demand_로 사용하는 방법을 학습하고, reflection tokens에 의해 안내되는 controlled generation을 도입하여 생성 품질과 기여도를 더욱 개선합니다.
입력에 앞에 붙는 검색된 구절을 사전 검색하거나, 검색기와 LM을 공동으로 사전 학습시킨 다음 작업 데이터 세트에 대한 몇 번의 샷 미세 조정(Izacard et al., 2022b)을 수행합니다. 이전 작업에서는 처음에 한 번만 검색하는 경우가 많았지만, Jiang et al. (2023)은 독점 LLM 위에 생성을 위해 적응적으로 구절을 검색하거나 Schick et al. (2023)은 명명된 엔티티에 대한 API 호출을 생성하는 LM을 훈련할 것을 제안합니다. 그러나 이러한 접근 방식의 향상된 작업 성능은 종종 런타임 효율성(Mallen et al., 2023), 관련 없는 컨텍스트에 대한 견고성(Shi et al., 2023), 기여도 부족(Liu et al., 2023; Gao et al., 2023)의 희생으로 이어집니다. 다양한 지시어에 따라 검색을 _온디맨드_로 사용하는 방법을 학습하고, 반영 토큰에 의해 안내되는 제어된 생성을 도입하여 생성 품질과 기여도를 더욱 개선합니다.