mkurman/phi4-MedIT-10B-o1

Phi-4 MedIT 10B o1 Model Card

Original model technical report

Phi-4 Technical Report

Phi-4-MedIT-10B-o1 🚀

Phi-4-MedIT-10B-o1 is a specialized large language model pruned for efficiency and performance:

Key Features:

• Pruned Model:

  • Derived from the phi-4 14B model, reduced to 10B parameters using the MKA2G pruning technique.
  • For details on the pruning methodology, see MedITSolutionsKurman/llama-pruning on GitHub.

• Reasoning SFT:

  • Fine-tuned with a single epoch of reasoning-specific supervised fine-tuning (SFT) for optimized reasoning tasks.

Recommended Usage:

• ReasonFlow Integration:

  • Achieve advanced reasoning by leveraging the ReasonFlow framework, which uses Multi-Path Generation with Noise.

• Monte-Carlo Tree Search (MCTS):

  • Use with MCTS-PyTorch to explore tree-based decision-making and inference.

Applications:

This model is ideal for:

• Efficient, high-quality reasoning tasks.
• Scenarios requiring enhanced performance in pruned environments.
• Integration with advanced frameworks for reasoning or decision-making.

Explore the possibilities with Phi-4-MedIT-10B-o1 and push the boundaries of efficient AI reasoning!

Safety

Approach

phi-4 has adopted a robust safety post-training approach. This approach leverages a variety of both open-source and in-house generated synthetic datasets. The overall technique employed to do the safety alignment is a combination of SFT (Supervised Fine-Tuning) and iterative DPO (Direct Preference Optimization), including publicly available datasets focusing on helpfulness and harmlessness as well as various questions and answers targeted to multiple safety categories.

Safety Evaluation and Red-Teaming

Prior to release, phi-4 followed a multi-faceted evaluation approach. Quantitative evaluation was conducted with multiple open-source safety benchmarks and in-house tools utilizing adversarial conversation simulation. For qualitative safety evaluation, we collaborated with the independent AI Red Team (AIRT) at Microsoft to assess safety risks posed by phi-4 in both average and adversarial user scenarios. In the average user scenario, AIRT emulated typical single-turn and multi-turn interactions to identify potentially risky behaviors. The adversarial user scenario tested a wide range of techniques aimed at intentionally subverting the model’s safety training including jailbreaks, encoding-based attacks, multi-turn attacks, and adversarial suffix attacks.

Please refer to the technical report for more details on safety alignment.

Model Quality

To understand the capabilities, we compare phi-4 with a set of models over OpenAI’s SimpleEval benchmark.

At the high-level overview of the model quality on representative benchmarks. For the table below, higher numbers indicate better performance:

Category	Benchmark	phi-4 (14B)	phi-3 (14B)	Qwen 2.5 (14B instruct)	GPT-4o-mini	Llama-3.3 (70B instruct)	Qwen 2.5 (72B instruct)	GPT-4o
Popular Aggregated Benchmark	MMLU	84.8	77.9	79.9	81.8	86.3	85.3	88.1
Science	GPQA	56.1	31.2	42.9	40.9	49.1	49.0	50.6
Math	MGSM MATH	80.6 80.4	53.5 44.6	79.6 75.6	86.5 73.0	89.1 66.3*	87.3 80.0	90.4 74.6
Code Generation	HumanEval	82.6	67.8	72.1	86.2	78.9*	80.4	90.6
Factual Knowledge	SimpleQA	3.0	7.6	5.4	9.9	20.9	10.2	39.4
Reasoning	DROP	75.5	68.3	85.5	79.3	90.2	76.7	80.9

* These scores are lower than those reported by Meta, perhaps because simple-evals has a strict formatting requirement that Llama models have particular trouble following. We use the simple-evals framework because it is reproducible, but Meta reports 77 for MATH and 88 for HumanEval on Llama-3.3-70B.

Usage

Input Formats

Given the nature of the training data, phi-4 is best suited for prompts using the chat format as follows:

<|im_start|>system<|im_sep|>
You are a medieval knight and must provide explanations to modern people.<|im_end|>
<|im_start|>user<|im_sep|>
How should I explain the Internet?<|im_end|>
<|im_start|>assistant<|im_sep|>

With transformers

import transformers

pipeline = transformers.pipeline(
    "text-generation",
    model="microsoft/phi-4",
    model_kwargs={"torch_dtype": "auto"},
    device_map="auto",
)

messages = [
    {"role": "system", "content": "You are a medieval knight and must provide explanations to modern people."},
    {"role": "user", "content": "How should I explain the Internet?"},
]

outputs = pipeline(messages, max_new_tokens=128)
print(outputs[0]["generated_text"][-1])

With ReasonFlow

Check it here: ReasonFlow framework

...
config = ReasonFlowConfig(
    num_of_thinkers=NUM_OF_THINKERS,
    num_of_thoughts=NUM_OF_THOUGHTS,
    topk_thinkers=TOPK_THINKERS,
)

index = 1

reason_flow = ReasonFlow(config, model, tokenizer)
output = reason_flow.generate_thoughts(chat_like_texts[index], 
                                       max_new_tokens=4096, 
                                       stream=True,
                                       device='cuda', 
                                       torch_dtype='bfloat16',
                                       do_sample=False, 
                                       temperature=None, 
                                       top_p=None, 
                                       top_k=None, 
                                       repetition_penalty=None)

With Monte-Carlo Tree Search

Check it here: MCTS-PyTorch

...
model.to("cuda")
input_tokens = input_ids['input_ids'].to('cuda') 

model.eval()

print(tokenizer.decode(input_tokens[0]))

while len(input_tokens[0]) < MAX_TOTAL_TOKENS:
    mcts = MCTS(model, tokenizer, max_depth=MAX_DEPTH, num_simulations=MAX_SIMULATIONS, temperature=TEMPERATURE, max_new_tokens=MAX_NEW_TOKENS, stop_tokens=model.config.eos_token_id)
    
    new_tokens = mcts.search(input_tokens)

    new_tokens = new_tokens[..., input_tokens.shape[-1]:]

    input_tokens = torch.cat([input_tokens, new_tokens], dim=-1)

    print(tokenizer.decode(new_tokens[0]), end='')

    if has_eos(new_tokens, eos_token=model.config.eos_token_id):
        break

Responsible AI Considerations

Like other language models, phi-4 can potentially behave in ways that are unfair, unreliable, or offensive. Some of the limiting behaviors to be aware of include:

• Quality of Service: The model is trained primarily on English text. Languages other than English will experience worse performance. English language varieties with less representation in the training data might experience worse performance than standard American English. phi-4 is not intended to support multilingual use.

• Representation of Harms & Perpetuation of Stereotypes: These models can over- or under-represent groups of people, erase representation of some groups, or reinforce demeaning or negative stereotypes. Despite safety post-training, these limitations may still be present due to differing levels of representation of different groups or prevalence of examples of negative stereotypes in training data that reflect real-world patterns and societal biases.

• Inappropriate or Offensive Content: These models may produce other types of inappropriate or offensive content, which may make it inappropriate to deploy for sensitive contexts without additional mitigations that are specific to the use case.

• Information Reliability: Language models can generate nonsensical content or fabricate content that might sound reasonable but is inaccurate or outdated.

• Limited Scope for Code: Majority of phi-4 training data is based in Python and uses common packages such as typing, math, random, collections, datetime, itertools. If the model generates Python scripts that utilize other packages or scripts in other languages, we strongly recommend users manually verify all API uses.

Developers should apply responsible AI best practices and are responsible for ensuring that a specific use case complies with relevant laws and regulations (e.g. privacy, trade, etc.). Using safety services like Azure AI Content Safety that have advanced guardrails is highly recommended. Important areas for consideration include:

• Allocation: Models may not be suitable for scenarios that could have consequential impact on legal status or the allocation of resources or life opportunities (ex: housing, employment, credit, etc.) without further assessments and additional debiasing techniques.

• High-Risk Scenarios: Developers should assess suitability of using models in high-risk scenarios where unfair, unreliable or offensive outputs might be extremely costly or lead to harm. This includes providing advice in sensitive or expert domains where accuracy and reliability are critical (ex: legal or health advice). Additional safeguards should be implemented at the application level according to the deployment context.

• Misinformation: Models may produce inaccurate information. Developers should follow transparency best practices and inform end-users they are interacting with an AI system. At the application level, developers can build feedback mechanisms and pipelines to ground responses in use-case specific, contextual information, a technique known as Retrieval Augmented Generation (RAG).

• Generation of Harmful Content: Developers should assess outputs for their context and use available safety classifiers or custom solutions appropriate for their use case.

• Misuse: Other forms of misuse such as fraud, spam, or malware production may be possible, and developers should ensure that their applications do not violate applicable laws and regulations.