Paper deep dive
ProtoCycle: Reflective Tool-Augmented Planning for Text-Guided Protein Design
Yutang Ge, Guojiang Zhao, Sihang Li, Zheng Cheng, Zifeng Zhao, Hanchen Xia, Guolin Ke, Linfeng Zhang, Zhifeng Gao, Yuguang Wang
Intelligence
Status: succeeded | Model: Gemma-4-26B-A4B | Prompt: intel-v1 | Confidence: 97%
Last extracted: 4/27/2026, 2:34:01 AM
Summary
ProtoCycle is an agentic framework for text-guided protein design that addresses the 'plan-execute gap' in generic LLMs. Instead of direct text-to-sequence generation, it uses an LLM as a high-level planner that interacts with a lightweight tool environment (scaffold generation, functional-site design, and evaluation) through a multi-round, feedback-driven cycle. The framework is optimized using a two-stage process: Supervised Fine-Tuning (SFT) and online Reinforcement Learning (RL) using Group Relative Policy Optimization (GRPO). Experimental results show that ProtoCycle-RL achieves superior language alignment (ProTrek score) and strong generalization to datasets like CAMEO compared to existing baselines like Pinal and ProDVa.
Entities (11)
Relation Signals (5)
Chai-1 â assesses â Foldability
confidence 100% · Foldability is assessed using Chai-1
LLM Planner â invokes â Scaffold Generation
confidence 100% · the planner outputs decisions... specifying which tool to invoke
Qwen2.5 7B â isbasefor â ProtoCycle
confidence 100% · We use Qwen2.5-7B as the base language model for our planner
ProtoCycle â trainedwith â GRPO
confidence 100% · update with Group Relative Policy Optimization (GRPO)
ProtoCycle â uses â LLM Planner
confidence 100% · ProtoCycle couples an LLM planner with a lightweight tool environment
Cypher Suggestions (0)
No Cypher suggestions yet.
Abstract
Abstract:Designing proteins that satisfy natural language functional requirements is a central goal in protein engineering. A straightforward baseline is to fine-tune generic instruction-tuned LLMs as direct text-to-sequence generators, but this is data- and compute-hungry. With limited supervision, LLMs can produce coherent plans in text yet fail to reliably realize them as sequences. This plan-execute gap motivates ProtoCycle, an agentic framework for protein design that uses LLMs primarily to drive a multi-round, feedback-driven decision cycle. ProtoCycle couples an LLM planner with a lightweight tool environment designed to emulate the iterative workflow of human protein engineering and uses LLM-driven reflection on tool feedback to revise plans. Trained with supervised trajectories and online reinforcement learning, ProtoCycle achieves strong language alignment while maintaining competitive foldability, and ablations show that reflection substantially improves sequence quality.
Tags
Links
- Source: https://arxiv.org/abs/2604.16896v1
- Canonical: https://arxiv.org/abs/2604.16896v1
Trouble viewing inline? Open PDF directly â
Full Text
83,437 characters extracted from source content.
Expand or collapse full text
ProtoCycle: Reflective Tool-Augmented Planning for Text-Guided Protein Design Yutang Ge 1,2 * Guojiang Zhao 2 * Sihang Li 3 Zheng Cheng 4 Zifeng Zhao 4 Hanchen Xia 1 Guolin Ke 2 Linfeng Zhang 2 Zhifeng Gao 2â Yuguang Wang 1â 1 Shanghai Jiao Tong University, School of Mathematical Sciences 2 DP Technology 3 University of Science and Technology of China 4 AI for Science Institute lhxn0627, yuguang.wang@sjtu.edu.cn, zhaogj, gaozf@dp.tech Abstract Designing proteins that satisfy natural language functional requirements is a central goal in protein engineering. A straightforward base- line is to fine-tune generic instruction-tuned LLMs as direct text-to-sequence generators, but this is data- and compute-hungry. With limited supervision, LLMs can produce coher- ent plans in text yet fail to reliably realize them as sequences. This planâexecute gap motivates ProtoCycle, an agentic framework for protein design that uses LLMs primarily to drive a multi-round, feedback-driven decision cycle. ProtoCycle couples an LLM planner with a lightweight tool environment designed to em- ulate the iterative workflow of human protein engineers and uses LLM-driven reflection on tool feedback to revise plans. Trained with supervised trajectories and online reinforce- ment learning, ProtoCycle achieves strong lan- guage alignment while maintaining competi- tive foldability, and ablations show that reflec- tion substantially improves sequence quality. Code and resources are available athttps: //github.com/huggggoooooo/ProtoCycle. 1 Introduction Designing proteins that achieve specified func- tional goals remains a central challenge with broad impact on enzyme engineering, therapeutics, and synthetic biology (Kortemme, 2024; Arnold, 2017; Ebrahimi and Samanta, 2023). Recent deep gen- erative models have already shown encouraging * Equal contribution. â Corresponding authors. progress toward this goal. On one hand, protein language models such as ProGen (Madani et al., 2023) and the ESM family (Rives et al., 2021; Lin et al., 2023) are pretrained on large sequence cor- pora and can be adapted to design tasks. On the other hand, text-guided design frameworks such as ProteinDT (Liu et al., 2025b), PAAG (Yuan et al., 2024), ProDVa (Liu et al., 2025a), and Pinal (Dai et al., 2025) explicitly condition on natural lan- guage to generate protein sequences. Because requirements are expressed in natural language while proteins are discrete amino-acid se- quences over a 20-letter alphabet, a straightforward baseline is to adapt a generic instruction-tuned LLM to map requirement text directly to protein sequences, following instruction-tuned generation in other domains (Zhao et al., 2025; Wei et al., 2021). However, our analysis in Section 2 reveals a consistent pattern: compared to direct sequence generation, LLMs appear stronger at planning in text. Under lightweight adaptation, they achieve higher plan quality and exhibit lower estimated uncertainty when producing plans, yet translating those plans into residue-level decisions that yield satisfying sequences remains challenging. This observation aligns with how protein design is carried out in practice: a protein engineer rarely goes from a specification to an ideal sequence in a single step. Instead, reaching a satisfying de- sign typically requires an iterative designâevaluateâ revise loop over multiple rounds. Concretely, prac- titioners typically start from an existing scaffold, make localized edits around functional sites, and arXiv:2604.16896v1 [q-bio.QM] 18 Apr 2026 evaluate candidates using structure- or function- related signals. When the outcome is unsatisfactory, they revisit earlier choices, adjust their strategy, and iterate (Wang et al., 2021; JĂ€ckel et al., 2008). Accordingly, we propose ProtoCycle, a text- guided protein generation framework that casts pro- tein design as an iterative cycle of planning, tool calling, evaluation, and revision. In ProtoCycle, the LLM primarily serves as a high-level planner, responsible for requirement decomposition, tool selection, strategy updates, reflection, and termi- nation decisions, while sequence generation and local editing are delegated to a lightweight set of specialized tools that can be called frequently and efficiently. We optimise the planner in this multi- step environment with a combination of supervised fine-tuning and reinforcement learning, so that it learns when to call which tools, how to interpret feedback, and how many cycles to run before ter- minating. Notably, ProtoCycle reaches competi- tive performance on our benchmarks using only âŒ2,500 training instances for planner optimization, whereas frontier end-to-end baselines such as Pinal are trained onâŒ1.7B proteinâtext pairs (Dai et al., 2025). This work makes the following contributions: âąWe empirically characterise the behaviour and limitations of generic LLMs for text-guided pro- tein generation. âąWe introduce a new paradigm for text-guided protein design by formulating it as a multi-round decision-making problem. âąWe propose ProtoCycle, an expert-workflow- inspired agent with a lightweight toolkit. âąWe empirically validate ProtoCycle and show via ablations that reflective revision improves sequence quality. 2 Diagnosing Generic Instruction-tuned LLMs as Direct text-guided Protein Generators This section studies a natural baseline: directly deploying generic instruction-tuned LLMs as zero- shot text-guided protein sequence generators. Our goal is to test whether their strong textual reasoning reliably translates into residue-level sequences. We probe this using a simple task and a token-level uncertainty estimator (Ma et al., 2025). Table 1: Qualitative interpretation of high/low aleatoric (AU) and epistemic (EU) uncertainty. LowHigh AUUnambiguous problemAmbiguous problem EUModel is confidentModel lacks knowledge 2.1 Problem Setup 2.1.1 Task: text-guided Protein Generation We use the protein-design subset of Mol- Instructions (Fang et al., 2024) as a benchmark, where each example pairs a natural-language de- scription with a target sequence. We run a data- scaling sanity check on Qwen2.5-7B (Qwen et al., 2025) and adopt a chain-of-thought prompting for- mat (Wei et al., 2023) that elicits a textual design rationale followed by an amino-acid sequence. We study three base models (Qwen2.5-7B, Qwen2.5-72B (Qwen et al., 2025), and Llama-3.1- 8B (Dubey et al., 2024)) under three usage modes: zero-shot, in-context learning (Brown et al., 2020), and supervised fine-tuning (SFT). 2.1.2 Token-level Uncertainty Estimation To probe model behaviour, we use Logits-induced Token Uncertainty (LogTokU) (Ma et al., 2025), which decomposes predictive uncertainty into: aleatoric uncertainty (AU), capturing ambiguity of the next-token distribution, and epistemic uncer- tainty (EU), capturing how weakly supported the prediction is by the modelâs evidence. Table 1 sum- marises the qualitative regimes for high and low AU and EU. Appendix A provides a token-level illustration (Figure 7) and full formulas. 2.2 Behaviour of Generic LLMs on text-guided Protein Generation 2.2.1 A data-scaling sanity check We fine-tune Qwen2.5-7B on subsets of the Mol-Instructions protein-design split (about200K protein-text pairs). For each run, we subsam- ple2K,20K,50K,100K, or the full200K exam- ples and fine-tune for two epochs, then evaluate textâsequence alignment on a held-out test set us- ing the ProTrek score (Su et al., 2025). Figure 1a shows that more data monotonically improves ProTrek, but gains are modest: increasing the dataset from2K to200K only lifts the score from roughly1to7, far below the ground-truth level ofâ 14.6. A simple power-law fit (Kaplan et al., 2020) suggests that reaching ProTrek around (a) Effect of fine-tuning data size.(b) Deviation of amino-acid composition of generated se- quences from ground truth. (c) Aleatoric and epistemic uncertainty on planning and sequence tokens. (d) Textâsequence alignment score.(e) Quality of textual design plans. Figure 1: Behaviour of generic instruction-tuned LLMs when directly used as text-guided protein generators. (a) Achieving strong textâsequence alignment would require very large data regimes. (b) Generated amino-acid compositions exhibit clear collapse, with SFT reducing these distortions to some extend. (c) Token-level uncertainty indicates that, compared to planning tokens, sequence tokens lie in a more evidence-poor regime. (d) Generic LLMs lag far behind ground truth, even after SFT. (e) In contrast, SFT substantially improves the quality of textual design plans, highlighting a gap between planning competence and sequence-generation ability. 12would require roughly6Ă 10 8 supervised ex- amples for Qwen2.5-7B, and approaching ground- truth scores requires on the order of10 9 pairs, sim- ilar in magnitude to Pinal (Dai et al., 2025), which trains a16B structure-conditioned model on1.7B proteinâtext pairs. 2.2.2 Sequence Collapse and Language Alignment Quality We next analyse behaviour under the chain-of- thought protocol where the model first outputs a textual plan and then a sequence. We begin with marginal amino-acid usage. For each model and usage mode, we aggregate gener- ated sequences and compare residue frequencies against those of the ground-truth sequences (see Appendix B.1). Figure 1b visualises the result- ing deviations. Generic LLMs exhibit pronounced distortions: some residues are heavily overused, while others rarely appear. SFT shifts these distri- butions toward the ground-truth profile, alleviating the most obvious collapse patterns. However, composition alone does not guarantee good designs. Figure 1d reports ProTrek scores as language alignment metric. All generic LLMs remain far below ground truth, even after SFT, indi- cating that better marginal statistics do not translate into strong language alignment. 2.2.3 Planning Quality We also evaluate the natural-language design ratio- nales produced before sequences. Plans are scored by DeepSeek-R1 (DeepSeek-AI et al., 2025) as an automatic judge; the rubric and its validation against expert ratings are given in Appendix B.2. Figure 1e shows that SFT yields clear and con- sistent improvements in plan quality across models and usage modes. Yet the corresponding sequence- level gains are small, suggesting that the models learn to describe what should be designed much better than they learn to implement these plans at the sequence level. 2.2.4 Uncertainty Asymmetry Figure 1c compares AU and EU on planning versus sequence tokens. AU is broadly similar in both re- gions, suggesting that, locally, predicting the next English word and the next residue are treated as similarly ambiguous. The asymmetry appears in EU: it is systematically higher on sequence tokens and aligns with the poor sequence quality, indicat- ing that models operate in an evidence-poor regime when mapping requirements to concrete amino- acid choices. Under our text-only SFT recipe, this pattern largely persists. SFT improves planning quality, but sequence-level structure and EU show no con- sistent gains. This supports a view in which SFT mainly refines a text-rich planning subspace, while providing too little new evidence to close the knowl- edge gap in the description-to-sequence mapping, so sequences become more format-plausible but remain functionally weak. 3 Method We now introduce ProtoCycle, which treats text- guided protein design as a multi-step decision pro- cess. In this view, LLM acts as a high-level planner that only decides how to design, while the actual sequence generation is delegated to a set of exter- nal tools (Yao et al., 2022; Schick et al., 2023). We first present an overview in Fig 2, then detail the planner interface and reflection mechanism. 3.1 Problem Formulation Formally, given a design requirementr, the agent interacts with a tool environmentTover multiple rounds. At roundt, the planner outputs decisions t and action a t : (a t ,s t ) = f (s tâ1 ,t tâ1 ,...,s 1 ,t 1 ,r),(1) wheret i = T (a i )denotes the tool feedback, which is a summary of tool results (see Section 3.3 for detail). Each action a i is factorised as a i = (a type i ,a arg i ), wherea type i specifies which tool to invoke, anda arg i specifies the input arguments for that tool. 3.2 Tool Environment To emulate the expert design process under realistic compute constraints, we instantiate a lightweight tool environmentTwith three corresponding tools: scaffold generation, functional-site design, and evaluation. Additional implementation details, query templates, and back-end mappings are pro- vided in Appendix C. Scaffold generation Given a free-form require- menttext, the scaffold generation tool retrieves and merges candidates from multiple protein knowledge bases (uni, 2025; Bansal et al., 2022; Blum et al., 2025; Binns et al., 2009), returningn scaffolds. Appendix C.1 shows more details. Functional-sitedesign. Givenaselected scaffoldand a local descriptiontext, the functional-site design tool generates site-level variants on top of ESM2-3B (Lin et al., 2023). Full procedure is shown in Appendix C.2. Evaluation. We evaluate candidates along two axes: Language Alignment and Foldability. Language Alignment is measured by ProTrek- 650M (Su et al., 2025). Foldability is assessed using Chai-1 (Chai Discovery, 2024) (no-MSA), reporting pTM, pLDDT, and PAE (Jumper et al., 2021), see Appendix F.2 for definitions. 3.3 Model Architecture Given the output(s t ,a t )at each step as defined in 1, the states t consists of two parts,<think>and <plan>, while the actiona t corresponds to a single <tool_call> token. At t = 1 (first round). âą <think> : the planner decomposes the require- mentrinto sub-requirementsr 1 ,r 2 ,...,r n , aiming to expose finer-grained design goals. âą <plan> : the planner maps eachr j to a tool configurationr j 7â a j = (type(r j ), arg(r j )), and proposes an execution ordera j 1 â a j 2 â ·â a j n . At tâ„ 2 (subsequent rounds). âą <think>: the planner summarizes the perfor- mance of the current strategy and reflects on the previous trajectorys tâ1 ,t tâ1 ,...,s 1 ,t 1 , deciding whether to (i) continue the current plan, (i) modify it, or (i) terminate. âą <plan>: the planner makes this high-level choice explicit, e.g., by executing the next scheduled sub-goalr j t or switching to an up- dated sub-goal r â j t . After generatings t (i.e.,<think>and<plan>), the planner produces the tool actiona t = (a type t ,a arg t ) . In practice, we serializea t into a JSON-like tag such as<tool_call> "name": Figure 2: Overview of ProtoCycle compared to a human protein engineer. Top: Human workflow, which iterates between scaffold selection, functional-site design, evaluation, and reflection until a satisfactory protein is obtained. Bottom: ProtoCycle, where a planner interacts with three tools (scaffold generation, functional-site design, and evaluation) via <think>/<plan>/<tool_call> steps and receives summarized tool feedback to revise its strategy. Figure 3: An example of reflection in ProtoCycle. a type t , "argument": a arg t </tool_call>. A lightweight runtime then parses this tag and in- vokes the corresponding tool. Figure 3 illustrates a concrete example attâ„ 2. We provide a full exam- ple trajectory in Appendix E.1.2, and a complete case study in Appendix I. Tool feedback summary.At each round of scaf- fold generation and functional-site design, the tools produce a set of sequencesP i = p i1 ,...,p im , which are scored by ProTrek-35M to obtainU i = u i1 ,...,u im . We compute (i) the number of sequencesm; (i) the current-round best score w i = max(U i ); (i) the global best scoreb iâ1 = max S iâ1 j=1 U j ; and (iv) the improvementâ i = w i â b iâ1 . The tool feedback summaryt i encapsulates these statistics and serves as a compact signal of the quality ofP i , which the planner then uses to decide how to proceed in the next round. Termination mechanism.When the planner de- cides to stop further tool use and prepare a final sequence, it triggers the evaluation tool to re-score the current top-kcandidates (we usek=5). The evaluation results are returned to the planner, which then either terminates and outputs the best-scoring sequence or resumes planning and invokes addi- tional tool calls to further refine the design when improvements are still needed. 3.4 Training Following recent cold-start + reinforcement learn- ing (RL) recipes (Guo et al., 2025; Zhao et al., 2025; Li et al., 2025; Wei et al., 2025), we train Pro- toCycle in two stages: supervised fine-tuning (SFT) to learn the basic<think>/<plan>/<tool_call> protocol and tool-usage conventions, followed by online RL in the tool environment of Section 3.2. Our training pipeline is implemented with the Ope- nAgentRL framework (Yu et al., 2025). 3.4.1 Supervised Fine-tuning We first train the planner with supervised fine- tuning on trajectories collected in the tool envi- ronment. Each training example is a sequence (r,s 1 ,t 1 ,s 2 ,t 2 ,...,s n ,t n ). Formally, the SFT objective for one trajectory is L SFT (r,s 1:n ,t 1:n ) =â 1 n n X i=1 logp Ξ s i | r,s <i ,t <i , which corresponds to a standard cross-entropy loss applied only on the planner states s 1 ,...,s n . 3.4.2 Online Reinforcement Learning After SFT, we further optimise the planner with online reinforcement learning in the real tool en- vironment. We treat the planner as a stochastic policy and update it with Group Relative Policy Optimization (GRPO) (Shao et al., 2024). The rewardR(Ï )is a shaped signal that com- bines several components, encouraging (i) well- formed planner outputs, (i) reasonable tool usage, (i) reflection after poor intermediate feedback, and (iv) solving the task in a moderate number of rounds. The exact form and coefficients of these terms are given in Appendix D. 4 Experiments 4.1 Design from textual descriptions DatasetWe use the protein-design subset of Mol- Instructions (Fang et al., 2024). We evaluate on three disjoint held-out test splits of 100 instances each (Eval-A, Eval-B, Eval-C) sampled from the dataset. We report main results averaged over the three splits, and provide split-wise results in Ap- pendix G. For SFT, we train on a disjoint set of 2,000 instances (see Appendix E.1 for details). For RL, we train for 5 epochs with 100 episodes per epoch, sampling requirements uniformly from the non-test pool in the real tool environment. Training details are in Appendix E. Baselines We use Qwen2.5-7B as the base lan- guage model for our planner and refer to the full agent (SFT + RL) as ProtoCycle. As generic LLM baselines in the same tool environment (Sec- tion 3.2), we instantiate tool-interacting agents whose planners are Qwen2.5-7B, Qwen3-8B, and Qwen2.5-72B (Qwen et al., 2025), an example of the full prompting format is provided in Ap- pendix E.1.3. We further compare against two text- guided protein design methods: Pinal (Dai et al., 2025), a large two-stage structure-conditioned model trained on about 1.7B proteinâtext pairs, and ProDVa (Liu et al., 2025a), which couples a text encoder and a protein LM with fragment retrieval. To isolate the effect of RL, we report both the supervised-only planner (ProtoCycle-SFT) and the planner after SFT followed by online RL (ProtoCycle-RL). Metrics We evaluate models using three groups of metrics (formal definitions in Appendix F), largely following prior work on text-guided pro- tein design (Kuang et al., 2025; Liu et al., 2025a). Sequence plausibility is measured by perplexity (PPL) (Jelinek et al., 1977) under ESM2-3B (Lin et al., 2023) and the percentage of repeated residues (Repeat) following PDFBench (Kuang et al., 2025) Foldability is assessed using predicted TM-score (pTM), mean per-residue confidence (pLDDT), and mean predicted aligned error (PAE) (Jumper et al., 2021) from Chai-1 (Chai Discovery, 2024), sum- marizing how likely a sequence is to fold into a sta- ble three-dimensional structure. Language align- ment is evaluated with the ProTrek score (Su et al., 2025), the EvoLLaMA score (Liu et al., 2024), and retrieval accuracy (Kuang et al., 2025), which quan- tify how well the designed sequences match their textual requirements in a joint embedding space. ResultsTable 2 compares generic LLM planners, prior text-guided protein design methods, and our ProtoCycle variants on sequence plausibility, fold- ability, and language alignment. Our key observa- tions are as follows. (1) ProtoCycle-RL is compet- itive with or better than specialized text-guided baselines. It achieves the strongest language align- ment overall, improving ProTrek by 3.66% over Pinal and by 21.97% over ProDVa, while main- taining competitive foldability: compared to Pinal, pTM/pLDDT drop by 2.15%/0.36% and PAE in- creases by 9.98%, while compared to ProDVa we improve all foldability metrics. (2) Online RL yeilds substantial improvements over SFT, es- pecially on language alignment. Compared to ProtoCycle-SFT, ProtoCycle-RL improves ProTrek by about 17.43% and increases retrieval accuracy by 11.43%. Meanwhile, RL also improves plau- sibility, reducing PPL and Repeat by 6.85% and 12.16%, respectively. Model Sequence PlausibilityFoldabilityLanguage Alignment PPLâRepeatâpTMâpLDDTâPAEâProTrekâEvoLLaMAâRetrievalâ Natural4.7372.1290.7620.8159.44314.6280.3280.848 Qwen2.5-7B-Agent8.2355.1530.5420.69915.2996.9260.2610.523 Qwen2.5-72B-Agent7.4145.3410.6180.71413.3438.7910.2670.563 Qwen3-8B-Agent7.2273.7950.6500.72313.4938.7050.2770.573 ProDVa5.2651.5800.7650.8008.76112.0370.3170.730 Pinal3.9909.3170.7920.8257.76814.1620.3180.807 ProtoCycle-SFT (ours)4.1492.9020.7340.80710.20012.5020.3170.840 ProtoCycle-RL (ours)3.8652.5490.7750.8228.54314.6810.3230.936 Table 2: Mol-Instructions protein design results avg. over three test splits (best, second-best). ModelTrain pLDDTâ ProTrekâ Kw. Rec.â Naturalâ0.7910.021.0 Pinalâ0.7511.780.39 ProDVa(Mol-Inst)Ă0.784.650.17 ProDVa(CAMEO)â0.8211.050.36 ProtoCycle-SFTĂ0.789.080.48 ProtoCycle-RLĂ0.8011.170.59 Table3:GeneralizationtoCAMEO(Best, Second Best). Train indicates whether the model is trained on keyword-style data (â) or not (Ă). Kw. Rec. denotes keyword recovery. 4.2 Generalization to CAMEO Setup. We evaluate cross-dataset generalization on CAMEO (Haas et al., 2018), a keyword-style protein-design dataset (see an example in Table 11). All ProtoCycle variants use the same tool environ- ment and evaluation protocol as in Section 4.1, and are trained only on the Mol-Instructions protein- design subset (i.e., without any keyword-style sam- ples in the training data). We compare against Pinal, who is trained with keyword-style data (â) and ProDVa with/without keyword-style training data (â/Ă). In addition to the metrics in Section 4.1, we evaluate keyword recovery (Kw. Rec.), which measures how many of the reference function key- words in the input are recovered from the designed sequence via InterProScan (Jones et al., 2014). See Appendix F.3 for the full definition. Results. Table 3 summarizes the main metrics. Although ProtoCycle is not trained to interpret com- pact keyword lists as requirements, it still transfers well to CAMEO. ProtoCycle-RL achieves strong language alignment and foldability, with perfor- mance comparable to Pinal, whose train dataset contains 800M keyword-protein pairs and ProDVa (CAMEO) ,who is trained directly on the CAMEO ToolProTrekâ PAEâ pLDDTâ Scaffold search11.428.960.83 Functional-site design12.8710.730.80 Table 4: Tool output quality when each tool is invoked once with arguments proposed by the trained planner. ComponentLatencyOutput Scaffold search4 s/round N retrieved sequences Functional-site design20 s/seq1 edited sequence Eval (ProTrek-35M)3 s/roundscores for N seqs Eval (ProTrek-650M)40 s/roundscores for N seqs Table 5: Wall-clock latency of each component. subset with 391M training examples, demonstrat- ing robust cross-format generalization. Full metrics are reported in Table 10. 4.3 Tool Efficiency SetupWe quantify the effectiveness and runtime of our tool environment by invoking each tool in isolation with arguments produced by the trained ProtoCycle planner. We report standard foldabil- ity and alignment metrics on the tool outputs, and measure wall-clock latency per invocation on our implementation. Results Table 4 shows that, under effective prompts, functional-site design improves language alignment relative to the raw scaffold-search output while keeping foldability metrics broadly compara- ble, suggesting that the tool can make meaningful local edits without catastrophically degrading struc- tural confidence. Table 5 shows that scaffold search takes a few seconds per round (4 s/round), whereas functional-site design is the dominant per-sequence cost (20 s/seq); during interaction we therefore use ProTrek-35M for fast per-round feedback and re- serve ProTrek-650M for final reporting. 4.4 Ablation: Reflection Mechanism SetupTo isolate the effect of reflection, we com- pare three planner variants under the same tool environment and evaluation protocol as in Sec- tion 4.1. (i) Workflow: we hard-code the tool order (SCAFFOLDâLOCAL-DESIGNâEVAL) in the prompt. (i) SFT (no reflection): the planner outputs<think>and<plan>but does not perform explicit reflection on previous plans; we fine-tune Qwen2.5-7B on non-reflective trajectories to learn the format and basic analysis. (i) ProtoCycle: our reflective planner, reported as ProtoCycle-SFT and ProtoCycle-RL (SFT+RL). Figure 4: Reflection ablation on final plausibility, fold- ability, and alignment metrics. Overall sequence quality Figure 4 shows that incorporating reflection yields substantially better designs. Reflective variants achieve nearly twice the language alignment score compared to their non-reflective counterparts, and also deliver a sig- nificant improvement in foldability metrics. Notably, SFT (no reflection) behaves similarly to the fixed Workflow, suggesting that learning the <think>/<plan>format and performing one-shot analysis alone is insufficient; without the ability to reflect on tool feedback and revise strategy, the planner largely degenerates into executing a me- chanical pipeline. Effect of reflection across interaction stepsTo understand how reflection helps, we track success rate as a function of the interaction step. For in- stancenat stept, letsucc(t) = 1 m P m n=1 succ n (t), (a) Success rate vs. step.(b) Valid / improving tool calls. Figure 5: Reflection improves step-wise optimization and tool-use efficiency. (a) Net improvement over the first step. (b) Wasted steps after the best score. Figure 6: Reflection improves decision quality and compute-aware stopping. wheresucc n (t) = 1[P â n (t)â„ 12],P â n (t) = max S t i=1 P n (i) , P n (i)is the set of ProTrek scores of all candidates produced at stepi, and mis the number of test instances. As shown in Fig- ure 5a, reflection increases success rates by roughly 25-40% over non-reflective baselines. Moreover, ProtoCycle (with RL) continues to improve with more steps, consistent with learning to explore and revise strategies based on intermediate feedback rather than repeatedly executing a static plan. We further quantify tool-use quality by count- ing (i) valid tool calls that execute successfully and return non-empty outputs, and (i) improving calls that strictly increase the current best score. Figure 5b shows that reflection increases both the valid-call rate (about+20%relative) and, more notably, the improving-call rate (about+40%rel- ative), suggesting that reflection helps the planner not only invoke tools correctly but also choose ar- guments that lead to measurable progress. Reflection induces more rational decisions.Fi- nally, we measure whether the planner makes de- cisions that improve upon the initial tool outcome. For each instancen, we compute the net gain P â n â P â n (1), whereP â n = max t P â n (t). Figure 6a shows that without reflection the net improvement is small, whereas reflectionâespecially with RLâ yields substantially larger gains (e.g., nearly a6Ă improvement over the Workflow baseline in our results). This indicates that reflective planners are better at turning intermediate feedback into con- crete strategy revisions. We also examine whether the planner stops once further tool calls become unproductive. Lett best (n)=arg max t P â n (t)and define waste(n) = t term (n) â t best (n) , wheret term is the terminal step. As shown in Figure 6b, reflection reduces wasted steps, indicating that the planner can recognize diminishing returns and terminate more promptly instead of continuing a fixed work- flow. 5 Conclusion Motivated by human protein-engineering work- flows and our finding that generic LLMs are more reliable planners than residue-level generators, we propose ProtoCycle, which couples an LLM plan- ner with a lightweight tool environment and an reflection mechanism for feedback-driven replan- ning. With only modest supervision and online interaction, ProtoCycle achieves foldability com- parable to current state-of-the-art methods while outperforming them on language alignment. Limitations One limitation of ProtoCycle is that our cur- rent functional-site design tool is intentionally lightweight to fit realistic compute budgets, so it can improve the odds of finding plausible candi- dates but cannot guarantee an "ideal" sequence that realizes the intended binding/catalytic geometry. This is consistent with broader findings in pro- tein design: while de novo design has advanced rapidly, reliably designing complex function (espe- cially high-specificity binding/catalysis) remains challenging and often still relies on iterative opti- mization and/or experimental screening rather than strict guarantees. Another limitation is the throughputâquality trade-off inherent to an agentic workflow. Because ProtoCycle invokes structure-informed tools during planning and evaluation, it incurs higher wall-clock time and compute than one-shot generators. Ethics Statement This work studies an agentic framework for text- guided protein sequence design. We do not use or collect personal user data, and our experiments rely on publicly available protein resources and bench- marks (including Mol-Instructions and CAMEO). We follow the original dataset and database li- censes/terms to ensure copyright- and license- compliant use, and we apply basic quality control (e.g., validity checks and filtering) to reduce mal- formed or misleading samples. Human experts were involved in a limited way during dataset construction. Specifically, experts provided a small number of seed examples that were used to prompt an external language model to synthesize additional training instances, and they performed a brief manual spot-check (around 10 samples) to flag obvious errors or inconsistencies. We did not collect or report any personally identify- ing information about the experts, and we treat this involvement as a lightweight sanity check rather than a comprehensive human-subject study. We acknowledge that synthetic data and bench- mark datasets can contain noise and systematic biases (e.g., uneven coverage across organisms, protein families, or functional annotations), which may propagate into model behavior and affect the diversity and reliability of generated candidates. We therefore encourage careful downstream valida- tion and transparent reporting of failure cases when applying the method to new domains. Finally, protein design is potentially dual-use. Our work is intended for benign scientific research and does not provide wet-lab protocols or procedu- ral instructions for harmful applications. Any real- world deployment should follow institutional over- sight and applicable biosecurity policies, including appropriate screening and usage constraints. References 2025. Uniprot: the universal protein knowledgebase in 2025. Nucleic acids research, 53(D1):D609âD617. Josh Abramson, Jonas Adler, Jack Dunger, Richard Evans, Tim Green, Alexander Pritzel, Olaf Ron- neberger, Lindsay Willmore, Andrew J. Ballard, Joshua Bambrick, Sebastian W. Bodenstein, and 1 others. 2024. Accurate structure prediction of biomolecular interactions with AlphaFold 3. Nature, 630(8016):493â500. Frances H Arnold. 2017. Directed evolution: bringing new chemistry to life. Angewandte Chemie (Interna- tional Ed. in English), 57(16):4143. Parit Bansal, Anne Morgat, Kristian B Axelsen, Venkatesh Muthukrishnan, Elisabeth Coudert, Lucila Aimo, Nevila Hyka-Nouspikel, Elisabeth Gasteiger, Arnaud Kerhornou, Teresa Batista Neto, and 1 others. 2022. Rhea, the reaction knowledgebase in 2022. Nucleic acids research, 50(D1):D693âD700. David Binns, Emily Dimmer, Rachael Huntley, Daniel Barrell, Claire Oâdonovan, and Rolf Apweiler. 2009. Quickgo: a web-based tool for gene ontology search- ing. Bioinformatics, 25(22):3045â3046. Matthias Blum, Antonina Andreeva, Laise Cavalcanti Florentino, Sara Rocio Chuguransky, Tiago Grego, Emma Hobbs, Beatriz Lazaro Pinto, Ailsa Orr, Ty- phaine Paysan-Lafosse, Irina Ponamareva, and 1 oth- ers. 2025. Interpro: the protein sequence classifica- tion resource in 2025<? mode longmeta?>. Nucleic acids research, 53(D1):D444âD456. Tom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah, Jared Kaplan, Prafulla Dhariwal, Arvind Neelakantan, Pranav Shyam, Girish Sastry, Amanda Askell, Sandhini Agarwal, Ariel Herbert-Voss, Gretchen Krueger, Tom Henighan, Rewon Child, Aditya Ramesh, Daniel M. Ziegler, Jeffrey Wu, Clemens Winter, and 12 others. 2020. Language Models are Few-Shot Learners. arXiv preprint. ArXiv:2005.14165. Chai Discovery. 2024. Chai-1: Decoding the molecular interactions of life. bioRxiv. Fengyuan Dai, Shiyang You, Yudian Zhu, Yuan Gao, Lihao Fu, Xibin Zhou, Jin Su, Chentong Wang, Yu- liang Fan, Xiaoxiao Ma, Xianjun Deng, Letong Yu, Hui Qian, Yan He, Yitao Ke, Chenchen Han, Xing Chang, Liangzhen Zheng, Sheng Wang, and 7 others. 2025. Toward de novo protein design from natural language. bioRxiv. DeepSeek-AI, Daya Guo, Dejian Yang, Haowei Zhang, Junxiao Song, Ruoyu Zhang, Runxin Xu, Qihao Zhu, Shirong Ma, Peiyi Wang, Xiao Bi, Xiaokang Zhang, Xingkai Yu, Yu Wu, Z. F. Wu, Zhibin Gou, Zhihong Shao, Zhuoshu Li, Ziyi Gao, and 181 others. 2025. DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning. arXiv preprint. ArXiv:2501.12948. Abhimanyu Dubey and 1 others. 2024. The llama 3 herd of models. Preprint, arXiv:2407.21783. Sasha B Ebrahimi and Devleena Samanta. 2023. En- gineering protein-based therapeutics through struc- tural and chemical design. Nature communications, 14(1):2411. Yin Fang, Xiaozhuan Liang, Ningyu Zhang, Kangwei Liu, Rui Huang, Zhuo Chen, Xiaohui Fan, and Hua- jun Chen. 2024. Mol-Instructions: A Large-Scale Biomolecular Instruction Dataset for Large Language Models. arXiv preprint. ArXiv:2306.08018. Yu Gu, Robert Tinn, Hao Cheng, Michael Lucas, Naoto Usuyama, Xiaodong Liu, Tristan Naumann, Jianfeng Gao, and Hoifung Poon. 2021. Domain-specific lan- guage model pretraining for biomedical natural lan- guage processing. ACM Transactions on Computing for Healthcare, 3(1):1â23. Daya Guo, Dejian Yang, Haowei Zhang, Junxiao Song, Peiyi Wang, Qihao Zhu, Runxin Xu, Ruoyu Zhang, Shirong Ma, Xiao Bi, and 1 others. 2025. Deepseek- r1 incentivizes reasoning in llms through reinforce- ment learning. Nature, 645(8081):633â638. JĂŒrgen Haas, Alessandro Barbato, Dario Behringer, Gabriel Studer, Steven Roth, Martino Bertoni, Khaled Mostaguir, Rafal Gumienny, and Torsten Schwede. 2018. Continuous automated model evalu- ation (cameo) complementing the critical assessment of structure prediction in casp12. Proteins: Structure, Function, and Bioinformatics, 86:387â398. Fred Jelinek, Robert L Mercer, Lalit R Bahl, and James K Baker. 1977. Perplexityâa measure of the difficulty of speech recognition tasks. The Journal of the Acoustical Society of America, 62(S1):S63âS63. Philip Jones, David Binns, Hsin-Yu Chang, Matthew Fraser, Weizhong Li, Craig McAnulla, Hamish McWilliam, John Maslen, Alex Mitchell, Gift Nuka, and 1 others. 2014.Interproscan 5: genome- scale protein function classification. Bioinformatics, 30(9):1236â1240. John Jumper, Richard Evans, Alexander Pritzel, Tim Green, Michael Figurnov, Olaf Ronneberger, Kathryn Tunyasuvunakool, Russ Bates, Augustin ĆœĂdek, Anna Potapenko, Alex Bridgland, Clemens Meyer, Simon A. A. Kohl, Andrew J. Ballard, Andrew Cowie, Bernardino Romera-Paredes, Stanislav Nikolov, Rishub Jain, Jonas Adler, and 15 others. 2021. Highly accurate protein structure prediction with Al- phaFold. Nature, 596(7873):583â589. Christian JĂ€ckel, Peter Kast, and Donald Hilvert. 2008. Protein Design by Directed Evolution. Annual Re- view of Biophysics, 37(1):153â173. Jared Kaplan, Sam McCandlish, Tom Henighan, Tom B. Brown, Benjamin Chess, Rewon Child, Scott Gray, Alec Radford, Jeffrey Wu, and Dario Amodei. 2020. Scaling Laws for Neural Language Models. arXiv preprint. ArXiv:2001.08361. Tanja Kortemme. 2024. De novo protein designâfrom new structures to programmable functions. Cell, 187(3):526â544. Jiahao Kuang, Nuowei Liu, Jie Wang, Changzhi Sun, Tao Ji, and Yuanbin Wu. 2025. Pdfbench: A bench- mark for de novo protein design from function. arXiv preprint arXiv:2505.20346. Jiatong Li, Weida Wang, Qinggang Zhang, Junxian Li, Di Zhang, Changmeng Zheng, Shufei Zhang, Xi- aoyong Wei, and Qing Li. 2025. Mol-r1: Towards explicit long-cot reasoning in molecule discovery. arXiv preprint arXiv:2508.08401. Zeming Lin, Halil Akin, Roshan Rao, Brian Hie, Zhongkai Zhu, Wenting Lu, Nikita Smetanin, Robert Verkuil, Ori Kabeli, Yaniv Shmueli, Allan Dos Santos Costa, Maryam Fazel-Zarandi, Tom Sercu, Salvatore Candido, and Alexander Rives. 2023. Evolutionary-scale prediction of atomic-level pro- tein structure with a language model. Science, 379(6637):1123â1130. Nuowei Liu, Jiahao Kuang, Yanting Liu, Tao Ji, Changzhi Sun, Man Lan, and Yuanbin Wu. 2025a. Protein Design with Dynamic Protein Vocabulary. arXiv preprint. ArXiv:2505.18966. Nuowei Liu, Changzhi Sun, Tao Ji, Junfeng Tian, Jianxin Tang, Yuanbin Wu, and Man Lan. 2024. Evol- lama: Enhancing llmsâ understanding of proteins via multimodal structure and sequence representations. arXiv preprint arXiv:2412.11618. Shengchao Liu, Yanjing Li, Zhuoxinran Li, Anthony Gitter, Yutao Zhu, Jiarui Lu, Zhao Xu, Weili Nie, Arvind Ramanathan, Chaowei Xiao, Jian Tang, Hongyu Guo, and Anima Anandkumar. 2025b. A text-guided protein design framework. Nature Ma- chine Intelligence, 7(4):580â591. Huan Ma, Jingdong Chen, Joey Tianyi Zhou, Guangyu Wang, and Changqing Zhang. 2025. Estimating LLM Uncertainty with Evidence. arXiv preprint. ArXiv:2502.00290. Ali Madani, Ben Krause, Eric R. Greene, Subu Subrama- nian, Benjamin P. Mohr, James M. Holton, Jose Luis Olmos, Caiming Xiong, Zachary Z. Sun, Richard Socher, James S. Fraser, and Nikhil Naik. 2023. Large language models generate functional protein sequences across diverse families. Nature Biotech- nology, 41(8):1099â1106. Valerio Mariani, Marco Biasini, Alessandro Bar- bato, and Torsten Schwede. 2013. lddt: a local superposition-free score for comparing protein struc- tures and models using distance difference tests. Bioinformatics, 29(21):2722â2728. Qwen, An Yang, Baosong Yang, Beichen Zhang, Binyuan Hui, Bo Zheng, Bowen Yu, Chengyuan Li, Dayiheng Liu, Fei Huang, Haoran Wei, Huan Lin, Jian Yang, Jianhong Tu, Jianwei Zhang, Jianxin Yang, Jiaxi Yang, Jingren Zhou, Junyang Lin, and 24 others. 2025. Qwen2.5 Technical Report. arXiv preprint. ArXiv:2412.15115. Alexander Rives, Joshua Meier, Tom Sercu, Siddharth Goyal, Zeming Lin, Jason Liu, Demi Guo, Myle Ott, C. Lawrence Zitnick, Jerry Ma, and Rob Fergus. 2021. Biological structure and function emerge from scaling unsupervised learning to 250 million protein sequences. Proceedings of the National Academy of Sciences, 118(15):e2016239118. Timo Schick, Jane Dwivedi-Yu, Roberto DessĂŹ, Roberta Raileanu, Maria Lomeli, Eric Hambro, Luke Zettle- moyer, Nicola Cancedda, and Thomas Scialom. 2023. Toolformer: Language models can teach themselves to use tools. Advances in Neural Information Pro- cessing Systems, 36:68539â68551. Zhihong Shao, Peiyi Wang, Qihao Zhu, Runxin Xu, Junxiao Song, Xiao Bi, Haowei Zhang, Mingchuan Zhang, Y. K. Li, Y. Wu, and Daya Guo. 2024. DeepSeekMath: Pushing the Limits of Mathemat- ical Reasoning in Open Language Models. arXiv preprint. ArXiv:2402.03300. Guangming Sheng, Chi Zhang, Zilingfeng Ye, Xibin Wu, Wang Zhang, Ru Zhang, Yanghua Peng, Haibin Lin, and Chuan Wu. 2024. Hybridflow: A flexible and efficient rlhf framework. arXiv preprint arXiv: 2409.19256. Jin Su, Yan He, Shiyang You, Shiyu Jiang, Xibin Zhou, Xuting Zhang, Yuxuan Wang, Xining Su, Igor Tol- stoy, Xing Chang, Hongyuan Lu, and Fajie Yuan. 2025. A trimodal protein language model enables advanced protein searches. Nature Biotechnology. Yajie Wang, Pu Xue, Mingfeng Cao, Tianhao Yu, Stephan T. Lane, and Huimin Zhao. 2021. Directed Evolution: Methodologies and Applications. Chemi- cal Reviews, 121(20):12384â12444. Jason Wei, Maarten Bosma, Vincent Y Zhao, Kelvin Guu, Adams Wei Yu, Brian Lester, Nan Du, An- drew M Dai, and Quoc V Le. 2021. Finetuned lan- guage models are zero-shot learners. arXiv preprint arXiv:2109.01652. Jason Wei, Xuezhi Wang, Dale Schuurmans, Maarten Bosma, Brian Ichter, Fei Xia, Ed Chi, Quoc Le, and Denny Zhou. 2023. Chain-of-Thought Prompting Elicits Reasoning in Large Language Models. arXiv preprint. ArXiv:2201.11903. Lai Wei, Yuting Li, Kaipeng Zheng, Chen Wang, Yue Wang, Linghe Kong, Lichao Sun, and Weiran Huang. 2025. Advancing multimodal reasoning via rein- forcement learning with cold start. arXiv preprint arXiv:2505.22334. Shunyu Yao, Jeffrey Zhao, Dian Yu, Nan Du, Izhak Shafran, Karthik R Narasimhan, and Yuan Cao. 2022. React: Synergizing reasoning and acting in language models. In The eleventh international conference on learning representations. Zhaochen Yu, Ling Yang, Jiaru Zou, Shuicheng Yan, and Mengdi Wang. 2025. Demystifying reinforce- ment learning in agentic reasoning. arXiv preprint arXiv:2510.11701. Chaohao Yuan, Songyou Li, Geyan Ye, Yikun Zhang, Long-Kai Huang, Wenbing Huang, Wei Liu, Jianhua Yao, and Yu Rong. 2024. Annotation-guided Protein Design with Multi-Level Domain Alignment. arXiv preprint. ArXiv:2404.16866. Yang Zhang and Jeffrey Skolnick. 2004. Scoring func- tion for automated assessment of protein structure template quality. Proteins: Structure, Function, and Bioinformatics, 57(4):702â710. Guojiang Zhao, Sihang Li, Zixiang Lu, Zheng Cheng, Haitao Lin, Lirong Wu, Hanchen Xia, Hengxing Cai, Wentao Guo, Hongshuai Wang, and 1 others. 2025. Molreasoner: Toward effective and inter- pretable reasoning for molecular llms. arXiv preprint arXiv:2508.02066. A Token-level Uncertainty Estimation with LogTokU We follow the Logits-induced Token Uncertainty (LogTokU) framework (Ma et al., 2025) to obtain a token-level uncertainty score for each generated token. At decoding stept, the language model produces logitsz t â R V over the vocabulary. At each decoding step the top-K logits are selected, S t =k 1 ,...,k K , z t,k = (z t ) k , k âS t , and treated as evidence over the small candidate setS t ; we adopt the same procedure in our experi- ments. Aleatoric uncertainty The evidence values are mapped to non-negative scalarse t,k â„ 0by apply- ing a rectified linear transform to the top-Klogits, e t,k = max 0,z t,k , and then used to define Dirichlet parameters α t,k = e t,k + 1, α t,0 = X kâS t α t,k . This induces a Dirichlet distribution over the categorical probabilities of theKcandidates. The aleatoric uncertainty (AU) at steptis defined as the expected entropy of this data distribution: AU t =â X kâS t α t,k α t,0 h Ï(α t,k + 1)â Ï(α t,0 + 1) i , whereÏ(·)is the digamma function. A largerAU t indicates that the next-token distribution is more diffuse or multi-modal (the problem itself is more ambiguous), whereas a smallAU t corresponds to an essentially unambiguous next token. Epistemic uncertainty The total amount of evi- dence, E t = X kâS t e t,k , is used to quantify how much experience the model has accumulated for the current context. The epis- temic uncertainty (EU) at step t is defined as EU t = K E t + K , so thatEU t is large when the total evidence is small (the model has little knowledge about this context), and decreases as the model accumulates more evidence. Token-level uncertainty.The token-level predic- tive uncertainty is defined as: u t = EU t AU t ,(2) Larger values ofu t correspond to tokens that are both weakly supported by the model (high epis- temic uncertainty) and not strongly constrained by the data (non-negligible aleatoric uncertainty), and are therefore treated as less reliable. B Additional Details for LLM Experiments B.1 Amino-acid Frequency Estimation Let the model generate sequencess 1 ,...,s N , wheres k = (a k1 ,...,a kL k )and each tokena kj is one of the 20 standard amino acids in the alpha- betA. For each residue typer â A, we estimate its empirical frequency under the model as f model (r) = P N k=1 P L k j=1 1[a kj = r] P N k=1 L k ,(3) and compute the corresponding ground-truth fre- quencyf gt (r)from the reference sequences in the dataset. In Figure 1b, we visualise the deviation âf (r) = f model (r)â f gt (r)for each residue type and model configuration. B.2 Plan-quality Scoring with DeepSeek-R1 We use DeepSeek-R1 (DeepSeek-AI et al., 2025) as an automatic judge to evaluate the quality of the textual design plans produced by the models. Each plan is scored along three dimensions: âąFramework: whether the overall design strategy is coherent and logically organised; âąBio: biological plausibility and correctness of the proposed operations; âąAlignment: how well the plan addresses the given textual requirement. The final planning quality score is computed as a weighted combination of these components: PlanScore = 0.4· framework + 0.3· bio + 0.3· alignment. To validate this automatic evaluation, we ran- domly select 20 descriptions and obtain human Figure 7: Token-level uncertainty visualization for a representative example. The top panel shows the generated explanation and sequence tokens, and the bottom colour bar encodes the scalar uncertaintyu t (Eq. 2) for each token. Regions with darker colours correspond to higher predictive uncertainty. expert ratings for both the framework and align- ment of the generated plans. The overall DeepSeek score shows a strong correlation with human judge- ments (Pearsonr = 0.73), and the alignment sub- score correlates with the ProTrek textâtext similar- ity between the plan and the requirement (Pearson r = 0.65). These results suggest that DeepSeek provides a reasonably faithful proxy for expert as- sessment of planning quality. C Tool Environment Details C.1 Scaffold Generation The scaffold-generation tool T (a SCAF , (text, num_sequences))wraps sev- eral database-specific queries behind a unified interface. The argumenttextis first lightly nor- malized (lowercasing and simple keyword match- ing) and then routed to one or more back-ends depending on its content: âągeneral functional descriptions (e.g., enzyme names, cofactor phrases)âUniProtKB (uni, 2025); âąpathway- or reaction-related cuesâ Rhea (Bansal et al., 2022); âą GO terms or GO-style phrasesâ QuickGO (Binns et al., 2009); âąexplicit domain / motif / structural keywordsâ InterPro (Blum et al., 2025). Each back-end returns a small set of candidate proteins together with basic annotations; we ex- tract their amino-acid sequences as candidate scaf- folds and merge them into a single list. When the union exceedsnum_sequences, we subsample using a simple ranking heuristic that favors pro- teins with richer annotations and lengths within a task-specific window. The mapping between query types and back-end databases is summarized in Table 6. Table 6: Query types and back-end resources used by the scaffold-generation tool. Query typeBack-end database General functional descriptionUniProtKB Pathway / reaction informationRhea GO terms or GO-style phrasesQuickGO Domain / structural motif cuesInterPro C.2 Implementation Details of the Functional-site Design Tool Identifying and locking annotated functional residues Given a scaffold sequences= (s 1 ,...,s L ), we query UniProt for residue-level an- notations across five feature types:binding_site, active_site,motif,domain, andregion. Each annotation returns an interval[i,j]on the sequence. We define the set of locked positions as L = [ (i,j)âA i,...,j, whereAis the union over all returned intervals. Residues inLare treated as immutable: subsequent motif insertions and local edits are restricted to indicest /âL. This preserves known functional or structurally critical regions while allowing the tool to operate on less constrained parts of the scaffold. Motif retrieval and ESM-based likelihood scor- ing The textual specificationtextdescribes the desired local function (e.g., "binds Zn 2+ with a Cys 2 His 2 coordination motif"). We usetext as a query to UniProt (uni, 2025) (optionally re- stricted by organism or keyword filters) and collect a small set of documented motif sequencesM = m (1) ,...,m (K) , each with length|m (k) | = â k . For a candidate motifmand an admissible inser- tion window[p,p + ââ 1]that does not intersect L, we construct a hypothetical edited sequence x (m,p) = (x (m,p) 1 ,...,x (m,p) L ), by replacing positionsp,...,p +ââ 1ofswithm (or insertingminto a loop if allowed by the back- bone representation). We then score this candidate using ESM2-3B (Lin et al., 2023). Because ESM2-3B is a masked language model, we estimate a pseudo negative log-likelihood (NLL) over a local region⊠(m,p) around the graft (e.g., the motif plus a small flanking window): NLL x (m,p) = â X tâ⊠(m,p) logp Ξ x (m,p) t x (m,p) , wherep Ξ (·| x )is the ESM2-3B conditional dis- tribution obtained by masking positiontand run- ning a single forward pass. Intuitively, this score penalizes candidates that are implausible under the protein language model, while leaving the rest of the scaffold unchanged. Across all motifsm â Mand all admissible insertion windowsp, the tool selects the configura- tion (m â ,p â ) = arg min m,p NLL x (m,p) , and returnsx (m â ,p â ) as the proposed functional-site design (subject to additional filtering below). D Reward Shaping Details Given a trajectoryÏwithTrounds, the final reward is R(Ï ) = ( R soft (Ï ), valid <answer> in Ï, â0.5,otherwise. The shaped component is R soft (Ï ) = 0.5R format (Ï ) + 1.0R tools (Ï ) + 0.5R ProTrek (Ï ) + 1.5R eff (T ) + R refl (Ï ). Format reward LetF tp (Ï )be1ifÏcontains at least one pair of<think>and<plan>tags, and0 otherwise. LetF ans (Ï )be1ifÏcontains a valid <answer> tag. We set R format (Ï ) = 0.5F tp (Ï ) + 0.5F ans (Ï ). Sequence-quality reward (ProTrek)LetP â (Ï ) be the best ProTrek score over all sequences inÏ, and letS(Ï )â0, 1indicate whether at least one sequence was generated. Then R ProTrek (Ï ) = ( clip P â (Ï )/20, 0, 1 , S(Ï ) = 1, â0.2,S(Ï ) = 0. Efficiency reward LetTbe the total number of rounds. We first separate the two regimes R eff (T ) = ( 0.1 (4â T ), T †4, r mid (T ), T > 4, and then define r mid (T ) =â 0.25 (T â 4)â 0.51[T > 8] â 1.01[T > 12]. Tool-usage reward For each roundi, we define a local tool rewardr tool (a i ,t i )and sum over the episode: R tools (Ï ) = T X i=1 r tool (a i ,t i ). In our implementation,r tool (a i ,t i )is (i) a small positive constantα gen = 0.3â0.6for the first few successful calls to generative tools, (i) an extra bonusα score = 0.3for successfulget_scorecalls, and (i) a penalty ofâ0.1 for failed tool calls. Reflection rewardLetbad(t i )â0, 1indicate that the feedback at roundiis below a threshold, and letrefl(s i+1 ) â 0, 1indicate that the next state contains an explicit reflection step. We define R refl (Ï ) = Tâ1 X i=1 r refl (t i ,s i+1 ), r refl (t i ,s i+1 ) = α good bad(t i ) refl(s i+1 ) + α noop bad(t i ) 1â refl(s i+1 ) , withα good > 0andα noop < 0(e.g.α good = 0.2 and α noop =â0.2). E Training Details This appendix provides the detailed training setup for ProtoCycle, including data construction, SFT settings, RL settings, tool-environment interface, and compute resources. E.1 SFT Trajectory Synthesis E.1.1 Overview We construct 2,000 multi-round tool-interaction tra- jectories for SFT by seeding GPT-4O with 5 expert- written trajectories as few-shot demonstrations. To mimic the reflection mechanism, we additionally prompt GPT-4O to condition its next-step decision on whether the previous tool call succeeded and on the returned ProTrek score: (i) if the tool call fails, it either retries with revised arguments or switches to an alternative tool; (i) if the tool call succeeds but the ProTrek score is below 12, it reflects on po- tential issues in the current workflow and replans; (i) if the tool call succeeds and the ProTrek score is above 12, it continues with the current plan. Each trajectory follows an interaction pattern with<think>,<plan>, and<tool_call>seg- ments, interleaved with tool observations. We ad- ditionally score the initial global plan of each syn- thesized trajectory using DEEPSEEK-R1 following the rubric in Appendix B.2; plans that do not meet the quality threshold are discarded and the trajec- tory is regenerated. E.1.2 Example Tool-Interaction Trajectories [USER REQUIREMENT] 1. The protein should have a specific function in intein-mediated protein splicing, cell-cell signaling and should be localized to plasma membrane, endoplasmic reticulum membrane, Golgi membrane in order to carry out this function efficiently. 2. The designed protein have Hint domains with specific sequence motifs that are crucial for its activity. [ASSISTANT] <think> [Requirement decomposition] - general function: "specific function in intein-mediated protein splicing, cell-cell signaling". - pathway: "intein-mediated protein splicing\", "cell-cell signaling" - domain: "Hint domains". - go: "plasma membrane", "endoplasmic reticulum membrane", "Golgi membrane". - motif: "specific sequence motifs that are crucial for its activity" - signal: localization to "plasma membrane, endoplasmic reticulum membrane, Golgi membrane </think> <plan> - First, explore functional scaffolds using pathway2seq with "intein-mediated protein splicing" as the pathway keyword, since it directly relates to the protein's function. - Once initial scaffolds are generated, introduce motif2constraints to enforce the presence of specific sequence motifs and signal2constraints to ensure proper localization. - After constraints are applied, call get_score to rank sequences and stop when the global best score is high and stable across multiple rounds. </plan> <tool_call> "name": "pathway2seq", "arguments": "text": "intein-mediated protein splicing" </tool_call> [TOOL OBSERVATION] Round: 1 - tool: pathway2seq - num_sequences_scored: 0 - delta_best: +0.000 - argument_pathway: intein-mediated protein splicing [ASSISTANT] <think> The first attempt to generate scaffolds using the pathway2seq tool with the pathway "intein-mediated protein splicing" did not yield any candidate sequences. This indicates that the argument may have been too specific or not aligned with the tool's expectations. To proceed, I will try a different approach by using the function2seq tool with a more general description of the protein's function. The phrase "intein-mediated protein splicing, cell-cell signaling" can be simplified to "intein-mediated splicing and cell signaling" to see if it retrieves any candidate sequences. This should help in generating a broader range of scaffolds that align with the protein's functional requirements. </think> <plan> Call the function2seq tool with the argument "intein-mediated splicing and cell signaling" to generate candidate protein sequences. </plan> <tool_call> "name": "function2seq", "arguments": "text": "intein-mediated splicing and cell signaling" </tool_call> [TOOL OBSERVATION] Round: 2 - tool: function2seq - num_sequences_scored: 50 - best_ProTrek_this_round: 13.858 - global_best_ProTrek: 13.858 - delta_best: +13.858 [ASSISTANT] <think> The function2seq tool successfully generated 20 candidate sequences, with the best ProTrek score reaching 13.858. This is a strong score, indicating that the scaffold aligns well with the functional requirements. However, the design still needs to incorporate specific constraints related to localization and sequence motifs. To ensure proper localization to the plasma membrane, endoplasmic reticulum membrane, and Golgi membrane, I will introduce motif2constraints to enforce the presence of specific sequence motifs crucial for the protein's activity. </think> <plan> Call the motif2constraints tool to enforce the presence of specific sequence motifs that are crucial for the protein's activity. </plan> <tool_call> "name": "motif2constraints", "arguments": "motif": "specific sequence motifs that are crucial for its activity" </tool_call> [TOOL OBSERVATION] Round: 3 - tool: motif2constraints - num_sequences_scored: 5 - global_best_ProTrek: 14.476 - prev_global_best_ProTrek: 13.858 - delta_best: +0.618 - argument_motif: specific sequence motifs that are crucial for its activity [ASSISTANT] <think> The motif constraint has improved the global ProTrek score, suggesting the current top candidate is already aligned with the requirement. Before allocating additional tool calls, I will run the evaluator once to obtain calibrated scores for language alignment and foldability of the current best candidate. </think> <plan> Call get_score to evaluate the current top candidates and verify requirement satisfaction. </plan> <tool_call> "name": "get_score", "arguments": "top_k": 5 </tool_call> [TOOL OBSERVATION] Round: 4 - tool: get_score - num_sequences_scored: 5 - ProTrek_best: 14.421 - pTM_best: 0.78 - pLDDT_best: 0.81 - PAE_best: 9.4 [ASSISTANT] <think> Given that the score is strong and stable, I will stop and return the current best candidate as the final design. </think> <answer> MDE...KSS </answer> E.1.3 Prompt for ProtoCycle In our implementation, tool-related instructions (i.e., the tool list, schemas, and calling formats) are automatically prepended to the model prompt by VERL (Sheng et al., 2024) based on the pro- vided tool specifications. Below we show the resulting trajectory prompt template used in our experiments. You are an agentic protein-design assistant. For each conversation, the user gives a natural-language design requirement for a protein, and you must explore tools step by step and finally output one amino-acid sequence. Conversation structure: - The user sends ONE main requirement description at the beginning. - Then the conversation alternates between: - assistant messages (you think/plan/call tools or give the final answer), - tool messages (OBSERVATION from the tools you called). Your message format: At every assistant turn you must choose EXACTLY ONE of the following patterns: (1) FIRST STEP (the very first assistant message after the user requirement), (2) INTERMEDIATE STEP (later steps that still call tools), (3) FINAL STEP (no more tool calls, only output the sequence). ================================ (1) FIRST ASSISTANT STEP ================================ The first assistant message in the conversation MUST have this structure: <think> [Requirement decomposition] - general function: present/not mentioned/not specified â quote key phrases if present. - pathway: ... - co-factor: ... - reaction: ... - domain: ... - dna-binding: ... - go: ... - motif: ... - signal: ... </think> <plan> - A high-level multi-step plan: * how you will explore scaffolds with Stage-1 tools, * when and why you will introduce constraints (Stage-2), * how you will use refinement/scoring tools (Stage-3), * under what conditions you will stop and output the final sequence. </plan> <tool_call> "name": "ONE_STAGE1_TOOL_NAME", "arguments": ... </tool_call> Rules for the first step: - You MUST call exactly ONE Stage-1 tool in the first step. - You MUST NOT call Stage-2 or Stage-3 tools in the first step. - You MUST NOT output <answer> in the first step. ================================ (2) INTERMEDIATE STEPS (LATER) ================================ Any later assistant step that still calls a tool MUST follow this structure: <think> - Summarize what has happened so far, especially the latest tool OBSERVATION (scores, whether sequences were found, whether constraints worked, etc.). - Decide whether to continue exploring scaffolds, add constraints, refine, or replan. - Choose exactly ONE tool to call next and explain briefly why it is appropriate now. - Explain how you choose its key arguments (e.g., simplify terms if previous calls failed). </think> <plan> - A concise description of the NEXT action: - which SINGLE tool you will call, - what main arguments you will pass, - and what you expect to learn or improve. </plan> <tool_call> "name": "TOOL_NAME", "arguments": ... </tool_call> Rules for intermediate steps: - You MUST include exactly one <think>, one <plan>, and one <tool_call>. - You MUST call exactly ONE tool per intermediate step. - You MUST NOT include <answer> in an intermediate step. ================================ (3) FINAL STEP (STOP AND OUTPUT SEQUENCE) ================================ Before the final step: - You MUST have called the scoring tool get_score at least once in this conversation. - You MUST read its OBSERVATION to know the current best design and its sequence. The FINAL assistant message MUST contain ONLY: <answer> A_SEQUENCE </answer> Rules for the final answer: - The content inside <answer> must be exactly ONE continuous amino-acid sequence (letters from ACDEFGHIKLMNPQRSTVWY), typically copied from the best sequence reported by the most recent get_score OBSERVATION. - Do NOT add any extra commentary, explanation, or text inside <answer>. - If the evaluation indicates that the current best candidate is not yet satisfactory (e.g., low alignment or poor foldability), you MUST NOT output <answer>. Instead, revise the plan and continue tool interaction. ================================ Heuristics and stopping criteria ================================ - Use several intermediate tool steps to: * generate scaffolds with Stage-1 tools, * optionally refine with Stage-2 constraints, * optionally refine with Stage-3 esm_inpaint, * and monitor the scores reported in OBSERVATIONs. - If the global ProTrek score is high, OR appears to have plateaued around a reasonable level, you SHOULD: 1) call get_score once to aggregate and re-score all known sequences, and then 2) in the NEXT turn, produce a FINAL STEP containing only <answer>. - If tools return OBSERVATIONs like "no new sequences" or num_sequences_scored=0, treat that call as FAILED: * in <think>, diagnose why (argument too long, wrong type, not a real motif/cofactor, etc.), * then adjust arguments (simplify or clean them) or switch to a more robust Stage-1 tool. Hard constraints: - NEVER call more than one tool in a single assistant message. - NEVER mix <answer> with <tool_call>. - ALWAYS use the FIRST STEP structure for the first assistant message, the INTERMEDIATE STEP structure for later tool-calling steps, and the FINAL STEP structure when you are ready to output the sequence. The following text is the design requirement you must satisfy for this conversation. requirement E.2 Supervised Fine-Tuning (SFT) We fine-tune QWEN2.5-7B-INSTRUCT (Qwen et al., 2025). The maximum training context length to 32,768 tokens; in practice, the multi-round tra- jectories typically occupy around 20K tokens. We use a total batch size of 32 and train for 5 epochs with full-parameter fine-tuning. Unless specified above, we follow the default SFT recipe in the OpenAgentRL framework (Yu et al., 2025) (e.g., optimizer and learning-rate schedule). E.3 Reinforcement Learning with GRPO Episode and rollout settings We enable multi- turn rollouts with a maximum of 8 user turns and 8 assistant turns. We set the maximum prompt length to 8,192 tokens and the maximum response length to 20,480 tokens. We samplen=4rollouts per requirement during training and usen=1for validation. For validation generation, we use top-p sampling with p=0.6 and temperature 1.0. Optimization settings We use an actor learning rate of1Ă 10 â6 , a training batch size of 64, and a PPO mini-batch size of 16. We aggregate the token- level loss by token-mean. We disable KL regular- ization (both in-reward KL and KL loss) and ap- ply clipped updates with clip ratios in[0.20, 0.28]. Gradient clipping is set to 1.0. For training data, each epoch contains 100 instances (disjoint from the test set), and we train for 5 epochs. E.4 Tool Environment Interface The tool environment and tool definitions are de- scribed in Section 3.2. Here we clarify the sum- mary returned to the planner after each tool call. Each tool returns a compact observation that in- cludes: (i) the tool name; (i) the number of se- quences scored in this round; (i) the global best ProTrek score so far; and (iv) the improvement over the previous global best. Concretely, we format the observation as: Round: t - tool: tool_name - num_sequences_scored: m - global_best_ProTrek: best - prev_global_best_ProTrek: prev_best - delta_best: best-prev_best - argument_type: arg E.5 Compute Resources All experiments are run on 8ĂA100 GPUs on a single node with CUDA 12.2, PyTorch 2.6.0, and VERL 0.5.0.dev0. F Evaluation Metrics F.1 Sequence Plausibility Pseudo-perplexity (PPL)We measure sequence plausibility using the pseudo-perplexity under a masked protein language model (ESM2-3B). Let Lbe the number of amino-acids. For each position i, we maskx i and compute the log-probability of the original token under the masked LM. PPL(x) = exp â 1 L L X i=1 logp Ξ (x i | x ) , wherex denotes the sequence with positioni masked. Repeat percentage (Repeat) We quantify lo- cal degeneration using the contiguous-repeat cri- terion, following PDFBench (Kuang et al., 2025). We scan window sizesw â 1,...,Wwhere W = min(20,ân/2â). For each start positioni, if the substringx i:i+w repeats consecutively at least three times, we mark the whole repeated span as a repeated region. LetR(x)be the union of all such repeated index intervals (merged if overlapping). We define Repeat(x) = 100· |R(x)| n . F.2 Foldability We evaluate foldability using the Chai-1 (Chai Dis- covery, 2024) structure predictor. For each se- quence, Chai-1 predicts a 3D structure and out- puts three confidence/error signals (pLDDT, PAE, pTM) following the standard AlphaFold-style defi- nitions (Jumper et al., 2021). pLDDT (predicted lDDT-Cα) The local dis- tance difference test (lDDT) is a superposition-free local quality metric (Mariani et al., 2013). Letd ij andd â ij be the CαâCαdistances between residues (i,j)in the predicted structure and the (unknown) true structure, respectively. For residuei, the lDDT- Cα is lDDT i = 1 |N (i)| X jâN (i) 1 4 X ÎŽââ 1 |d ij â d â ij | < ÎŽ , whereâ =0.5, 1, 2, 4,N (i)is a local neighbor- hood of residue pairs used by lDDT. Chai-1 outputs pLDDT i â E[lDDT i ]as a learned per-residue confidence score. PAE (predicted aligned error) Predicted aligned error (PAE) measures confidence in the relative placement of two residues/domains. Following AlphaFold, define the alignment-frame error e ij = T â1 i ⊠x j â (T â i ) â1 ⊠x â j , i.e., the positional error of residuejafter aligning predicted and true structures on residuei. Chai-1 outputs PAE ij â E[e ij ], which is typically reported in Ă and is not symmet- ric in general. pTM (predicted TM-score) TM-score is a global superposition metric (Zhang and Skolnick, 2004). AlphaFold derives a computable predictor using the aligned-error distribution: pTM = max i 1 N N X j=1 E ïŁź ïŁŻ ïŁ° 1 1 + e ij d 0 (N ) 2 ïŁč ïŁș ïŁ» , d 0 (N ) = 1.24 max(N, 19)â 15 1/3 â 1.8, whereNis the number of residues. Chai-1 reports pTM using the same underlying definition. F.3 Language Alignment ProTrek score We compute a textâsequence alignment score using ProTrek-650M (Su et al., 2025). In our implementation, we load the released ProTrek-650M checkpoint together with its accom- panying text/protein encoders (PubMedBERT (Gu et al., 2021) and ESM2-650M, respectively). Given a textual descriptiontand a protein sequencex, ProTrek encodes them into embeddingse t = Ï t (t) ande x = Ï p (x), and returns the temperature-scaled inner product: ProTrek(t,x) = e †x e t Ï , where Ï is the model temperature parameter. EvoLLaMA score Following (Kuang et al., 2025), we adopt a generative alignment metric based on EvoLLaMA (Liu et al., 2024). Given a sequencex, EvoLLaMA is prompted to generate a predicted function descriptiont âČ . We then embed the ground-truth texttand the generated textt âČ us- ing PubMedBERT (Gu et al., 2021), average token embeddings, and compute cosine similarity: EvoLLaMA(t,x) =sim ïŁ« ïŁ 1 |t| |t| X i=1 Embed(t i ), 1 |t âČ | |t âČ | X j=1 Embed(t âČ j ) ïŁ¶ ïŁž . Retrieval accuracyRetrieval accuracy evaluates whether a description retrieves its matched se- quence among randomly sampled negatives. For each test descriptiont k , we form a candidate setC k containing its matched sequencex k andK â 1 randomly sampled negative sequences (we use K = 32in all experiments), compute similarities in the same embedding space, and count a hit if the matched sequence ranks top-1: RA = 1 N N X k=1 1 arg max xâC k s(t k ,x) = x k . Keywordrecovery FollowingPDF- Bench (Kuang et al., 2025), we define keyword recovery as a protein-level metric based on InterProScan annotations. For each test instance, letK ref denote the reference set of function keywords provided in the input, and letK pred denote the set of function keywords identified from the designed sequence using InterProScan (Jones et al., 2014). Keyword recovery is then defined as Kw. Rec. = |K pred â© K ref | |K ref | . G Robustness Across Disjoint Test Splits To assess robustness to test-set sampling, we report split-wise results on three disjoint held-out splits (Eval-A/B/C). Tables 7,8,9 show that the relative ranking and overall trends are consistent across splits. Model Sequence PlausibilityFoldabilityLanguage Alignment PPLâRepeatâpTMâpLDDTâPAEâProTrekâEvoLLaMAâRetrievalâ Natural4.5832.6490.7570.8189.10914.5830.3270.875 Qwen2.5-7B-Agent8.2045.1800.5420.69915.2966.9380.2610.520 Qwen2.5-72B-Agent7.3715.3020.6180.71413.2498.8030.2670.560 Qwen3-8B-Agent7.2003.8460.6500.72313.4908.7040.2770.570 ProDVa5.1381.6580.7510.7988.89912.3940.3160.720 Pinal4.0689.1060.7900.8227.79714.1940.3170.830 ProtoCycle-SFT (ours)4.3914.1400.7430.8049.78012.1390.3220.800 ProtoCycle-RL (ours)3.7323.6870.7740.8168.52914.9610.3221.000 Table 7: Mol-Instructions protein design results on Eval-A (best, second-best). Model Sequence PlausibilityFoldabilityLanguage Alignment PPLâRepeatâpTMâpLDDTâPAEâProTrekâEvoLLaMAâRetrievalâ Natural4.8882.0510.7690.8039.97514.7350.3250.870 Qwen2.5-7B-Agent7.5205.9600.5200.68216.4206.3200.2480.460 Qwen2.5-72B-Agent7.0505.7400.6060.70113.8808.4200.2540.520 Qwen3-8B-Agent6.6204.4200.6280.70514.2808.1500.2620.510 ProDVa6.0611.7440.7780.7998.66111.5770.3260.710 Pinal3.9169.9450.7850.8218.13214.071 0.3240.820 ProtoCycle-SFT (ours)4.4552.1940.7170.79810.75212.7840.3230.838 ProtoCycle-RL (ours)3.9172.4240.7660.8128.95114.3950.3260.880 Table 8: Mol-Instructions protein design results on Eval-B (best, second-best). Model Sequence PlausibilityFoldabilityLanguage Alignment PPLâRepeatâpTMâpLDDTâPAEâProTrekâEvoLLaMAâRetrievalâ Natural4.7401.6860.7600.8239.24514.5660.3300.800 Qwen2.5-7B-Agent8.9804.3200.5650.71514.1807.5200.2740.590 Qwen2.5-72B-Agent7.8204.9800.6310.72612.9009.1500.2790.610 Qwen3-8B-Agent7.8603.1200.6720.74112.7109.2600.2920.640 ProDVa4.5951.3380.7670.8028.72512.1390.3100.760 Pinal3.9848.9000.8000.8337.37314.2190.3130.770 ProtoCycle-SFT (ours)3.6022.3730.7410.81810.06712.5820.3060.880 ProtoCycle-RL (ours)3.9461.5360.7850.8378.14814.6870.3200.929 Table 9: Mol-Instructions protein design results on Eval-C (best, second-best). ModelTrain Sequence PlausibilityFoldabilityLanguage Alignment PPLâRepeatâpTMâpLDDTâPAEâProTrekâK.w. Rec.âRetrievalâ Naturalâ4.3582.4160.7560.7888.63110.0201.0000.690 Pinalâ5.6959.4970.6830.7529.97411.7840.3940.550 ProDVaĂ6.3621.1470.7560.7768.6634.6460.1720.120 ProDVaâ5.6530.8270.7940.8176.97511.0540.3610.360 ProtoCycle-SFTĂ5.5932.1440.7060.77912.1299.0780.4800.684 ProtoCycle-RLĂ4.0612.1280.7560.80510.49311.1710.5920.833 Table 10: Full CAMEO evaluation metrics (Best,Second Best). Train indicates whether the model is trained on CAMEO (â) or not (Ă). H Additional Generalization Results on CAMEO H.1Input style comparison: Mol-Instructions vs. CAMEO Dataset Example input Mol-Instructions (protein design) (Fang et al., 2024) Create a protein sequence that satisfies the following specifications: 1. The protein must exhibit the following characteristics: Also acts as a cofactor with GATA4, a key cardiac regulator. 2. The protein should have metal ion binding and be active in nucleus. CAMEO (Haas et al., 2018) Generate a protein sequence for a novel protein that integrates the following function keywords: Polyphenol_oxidase_C, Tyrosinase_Cu-bd,TAT_signal Table 11: Representative conditioning-text examples from Mol-Instructions (protein-design subset) and CAMEO. To contextualize the cross-dataset generaliza- tion setting, Table 11 provides representative ex- amples of the conditioning-text formats in Mol- Instructions (protein design) (Fang et al., 2024) and CAMEO (Haas et al., 2018). Mol-Instructions typically uses instruction-style natural language requirements, whereas CAMEO uses compact keyword-like annotations. The examples are lightly paraphrased for readability. H.2 Full results We provide full evaluation metrics for CAMEO, including sequence plausibility, foldability, and lan- guage alignment. The evaluation protocol matches Section 4.1. ProtoCycle variants are trained on the Mol-Instructions protein-design subset only (i.e., no CAMEO supervision), and are tested directly on CAMEO. We report both ProDVa with/without keyword-style training (â/Ă), and Pinal with keyword-style training (â). Table 10 shows that ProtoCycle-RL substantially improves generalization to keyword-style inputs over its SFT-only variant. Compared to PINAL (trained on keyword-style data), ProtoCycle-RL achieves much higher keyword recovery (+50.3%) and retrieval (+51.5%), with markedly less repeti- tion, while its ProTrek is slightly lower (â5.2%). Against PRODVA trained only on Mol-Instructions, PROTOCYCLE-RL yields large gains in alignment; relative to PRODVA-CAMEO, PROTOCYCLE-RL attains higher keyword recovery and retrieval accu- racy, albeit with weaker foldability. I Case Study We select a representative example (full trajectory in I.2) to illustrate ProtoCycleâs multi-round deci- sion process (Fig. 8â9). In addition to language alignment and foldability metrics, we fold interme- diate sequences with AlphaFold3 (Abramson et al., 2024) and use InterProScan (Jones et al., 2014) to identify key functional regions, which are then annotated on the folded structure. The case highlights two behaviors. (i) Replan- ning under tool failure: ProtoCycle initially fol- lows a GO-term scaffold search plus functional- site refinement, but when the motif2constraints call fails in Round 3, it immediately switches to an alternative plan based on general-function scaf- fold search and continues refinement. (i) Plateau awareness and proper termination: after obtain- ing a high semantic-similarity scaffold (Round 5), further functional-site design does not improve the score, and ProtoCycle detects diminishing returns and stops with the best candidate. Figure 8: Case study (Part I) Figure 9: Case study (Part I) Figure 10: Pinal on the case-study requirement. Figure 11: ProtoCycle (final) on the same requirement. I.1 Comparison with baselines ProtoCycle and a strong baseline (Pinal) both achieve good language alignment and foldability on this case (Figs. 10 and 11). The requirement consists of three key aspects: (i) a stable heme b binding site, (i) oxygen transport/oxygen-binding function, and (i) association with the hemoglobin complex. In both designs, we observe evidence con- sistent with (i) via oxygen-transport related path- way/function annotations, and with (i) via heme binding annotations around the predicted pocket. However, for (i), Pinal mainly provides homology- level support (from GO/InterPro mappings) with- out residue-level interface site annotations, whereas ProtoCycle yields a design with clearer structure- level support for complex/interface-related regions highlighted by our annotation pipeline (Fig. 11). I.2 Full trajectory of Fig. 8â9 [USER REQUIREMENT] 1. The heme b binding site should be stable and able to withstand changes in environmental conditions. 2. For general function, the protein need meet that Involved in oxygen transport from the gills to the various peripheral tissues. 3. The protein can be targeted to hemoglobin complex and perform heme binding, oxygen binding. [ASSISTANT] <think> [Requirement decomposition] - general function: "Involved in oxygen transport from the gills to the various peripheral tissues." - pathway: "oxygen transport pathway." - co-factor: "heme b binding site." - go: "oxygen transport," "heme binding," "oxygen binding." </think> <plan> - First, explore functional scaffolds using go2seq with the GO-like terms "oxygen transport," "heme binding," and "oxygen binding." - Once promising scaffolds are identified, introduce constraints using motif2constraints and cofactor2constraints to ensure heme b binding and stability. - Finally, use get_score to rank and select the best-performing sequence, stopping when the global best score stabilizes across iterations. </plan> <tool_call> "name": "go2seq", "arguments": "go_term": "oxygen transport, heme binding, oxygen binding" </tool_call> [TOOL_OBSERVATION] Round: 1 - tool: go2seq - num_sequences_scored: 17 - best_ProTrek_this_round: 12.442 - global_best_ProTrek: 12.442 - delta_best: +12.442 [ASSISTANT] <think> The first round of scaffold generation using the go2seq tool was successful, yielding 17 new sequences with a global best ProTrek score of 12.442. This score indicates a promising start. Next, I will introduce constraints to refine the design further. The protein needs to bind heme b and be stable under varying environmental conditions. Therefore, I will use the cofactor2constraints tool to enforce the requirement for heme b binding. </think> <plan> - I will call the cofactor2constraints tool with the argument "cofactor": "heme b" to ensure that the generated sequences include features necessary for heme b binding. </plan> <tool_call> "name": "cofactor2constraints", "arguments": "cofactor": "heme b" </tool_call> [TOOL OBSERVATION] Round: 2 - tool: cofactor2constraints - num_sequences_scored: 5 - best_ProTrek_this_round: 14.274 - global_best_ProTrek: 14.274 - prev_global_best_ProTrek: 12.442 - delta_best: +1.832 [ASSISTANT] <think> The second round using the cofactor2constraints tool was successful, yielding 5 new sequences and improving the global best ProTrek score to 14.274 (+1.832). This indicates progress, to further refine the design, I will now introduce additional constraints related to the hemoglobin complex and the oxygen transport function. Since no specific motif was mentioned in the requirements, I will use the motif2constraints tool to enforce a general hemoglobin complex motif that could enhance stability and functionality. </think> <plan> I will call the motif2constraints tool with the argument "motif": "hemoglobin complex" to ensure that the generated sequences include features necessary for stable heme b binding. </plan> <tool_call> "name": "motif2constraints", "arguments": "motif": "hemoglobin complex" </tool_call> [TOOL OBSEVATION] Round: 3 - tool: motif2constraints - num_sequences_scored: 0 - global_best_ProTrek: 14.274 - prev_global_best_ProTrek: 14.274 - delta_best: +0.000 [ASSISTANT] <think> The attempt to introduce a hemoglobin complex motif using the motif2constraints tool did not yield any new usable sequences. This suggests that the motif argument may have been too vague or not specific enough to produce results. To address this, I will replan and try a different approach. Since the protein needs to be stable and function in oxygen transport, I will use the function2seq tool with a more focused description of the general function, specifically targeting "oxygen transport from gills to peripheral tissues." This may help generate new scaffolds that align better with the requirement. </think> <plan> I will call the function2seq tool with the argument "text": "oxygen transport from gills to peripheral tissues" to generate new scaffolds based on the general function of the protein. </plan> <tool_call> "name": "function2seq", "arguments": "text": "oxygen transport from gills to peripheral tissues" </tool_call> [TOOL OBSERVATION] Round: 4 - tool: function2seq - num_sequences_scored: 50 - best_ProTrek_this_round: 18.744 - global_best_ProTrek: 18.744 - prev_global_best_ProTrek: 14.274 - delta_best: +4.469 [ASSISTANT] <think> The fourth round using the function2seq tool was successful, yielding 20 new sequences and significantly improving the global best ProTrek score to 18.744 (+4.469), indicating that we are nearing a satisfactory design. To further refine the design and ensure stability and functionality, I will now apply the signal2constraints tool to introduce a signal peptide that targets the protein to the hemoglobin complex, which is essential for its function in oxygen transport. </think> <plan> I will call the signal2constraints tool with argument "hemoglobin complex" to enforce the requirement for a signal peptide that targets the protein to the hemoglobin complex. </plan> <tool_call> "name": "signal2constraints", "arguments": "signal": "hemoglobin complex" </tool_call> [TOOL OBSERVATION] Round: 5 - tool: signal2constraints - num_sequences_scored: 5 - best_ProTrek_this_round: 17.494 - global_best_ProTrek: 18.744 - prev_global_best_ProTrek: 18.744 - delta_best: -1.250 [ASSISTANT] <think> The fifth round using the signal2constraints tool successfully generated 5 new sequences, but the global best ProTrek score remains unchanged at 18.744. This indicates that while we have made progress, the design may have reached a plateau. Given that the global best score has stabilized, I will proceed to aggregate and re-score all known sequences using the get_score tool. This will allow us to identify the best design based on the current set of sequences and their scores. </think> <plan> I will call the get_score tool to aggregate and re-score all known sequences, identifying the best design based on the current scores. </plan> <tool_call> "name": "get_score", "arguments": "top_k": 5 </tool_call> [TOOL OBSERVATION] Round: 6 - tool: get_score - num_sequences_scored: 5 - ProTrek_best: 18.744 - pTM_best: 0.92 - pLDDT_best: 0.91 - PAE_best: 3.4 [ASSISTANT] <think> Given that the score is strong and stable, I will stop and return the current best candidate as the final design. </think> <answer> MVK...NYH </answer>