Skip to content

Yu-Yang-Li/StarWhisper

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

星语StarWhisper

🤖 GitHub项目 🥳 星语绘卷权重

GitHub Repo stars GitHub Code License GitHub last commit

在国家天文台人工智能工作组的支持下,我们开发了StarWhisper3天文大模型系列,包括语言模型、时序模型、多模态模型(7B-72B)。

将随着技术文章发表,逐渐开源相应模型权重、使用Demo。

版本更新:

1.通过清洗订正科普、科研数据飞轮得到的数据,改进训练方法,进一步提升了模型的天文物理、代码与Agent能力。

2.基于LLM、MLLM、Audio Model的天文时间序列处理,相关技术报告即将发布。

3.通过工具调用实现了多模态多任务框架、与望远镜控制系统进行对接。

功能展示

快速使用

下面是一个使用StarWhisper模型,进行多轮对话交互的样例:

from modelscope import AutoModelForCausalLM, AutoTokenizer
from modelscope import GenerationConfig

tokenizer = AutoTokenizer.from_pretrained("AstroYuYang/StarWhisper3", trust_remote_code=True)

model = AutoModelForCausalLM.from_pretrained("AstroYuYang/StarWhisper3", device_map="auto", trust_remote_code=True).eval()

# 在cpu上推理
# model = AutoModelForCausalLM.from_pretrained("AstroYuYang/StarWhisper", device_map="cpu", trust_remote_code=True).eval()

# model.generation_config = GenerationConfig.from_pretrained("AstroYuYang/StarWhisper", trust_remote_code=True) # 可指定不同超参

# 1st
response, history = model.chat(tokenizer, "你好", history=None)
print(response)

# 2nd
response, history = model.chat(tokenizer, "什么是黑洞?", history=history)
print(response)

司天工程

司天工程是我国天文学家面向时域天文学所提出的“十五五”天文重大基础设施,一期计划在国内多个优选观测台址布置54台(18组)口径1米级的大视场望远镜,组成多波段同时监测网络,每30分钟完成1万平方度天区的高精度三色“凝视”巡天。司天的采样频率比全球其它巡天项目高近两个量级,将突破目前探测时标的限制,在新的空域和时域下发现大批新天体、新现象,在宇宙极端高能爆发源、引力波电磁对应体、系外行星和太阳系天体等理论和观测研究中形成新的突破,在“两暗一黑三起源”等重大科学问题研究以及地球文明灾难预警等国家空间安全问题方面发挥重要作用。

其中司天"大脑"作为数据智能处理中枢,需要适配于天文的AI工具。StarWhisper作为其备选方案,在使用大模型整合天文知识的同时,探索多模态解决具体天文问题的可能性。

许可证信息

项目源码遵从Apache-2.0 license,Qwen1.5-14B Chat的模型权重使用需遵从相应许可。

To do list

大语言模型(科普方式)

  • 在相关材料上进行二次预训练,扩充天文知识。
  • 调整监督微调中,通用数据和专业数据的比例,缓解灾难性遗忘问题。
  • 通过人工反馈的强化学习,进一步提升模型性能。
  • 通过特定数据集微调,提升模型总结能力,进一步适配知识库。
  • 完成司天-变星知识图谱,与模型链接,进一步降低变星领域的幻觉现象。

专业多模态(科研工具)

  • 开源在多模态微调权重[即将开源]。
  • 进一步探索多模态模型在天文图像生成与识别上应用的可能性。

观测Agent(司天大脑)

  • 提升模型在天文领域的编程能力。
  • 在MiniSiTian/司天样机上,进行与天文环境交互的Agent探索工作。
  • 考虑通过工具学习,链接天文专业工具。
  • 尝试Agent相关工作,验证作为司天大脑备选方案的可行性。

引用

如果这篇工作对你有帮助,请引用:

@Misc{chatglm-for-variable-star,

title = {StarGLM},

author = {YuYang Li, CunShi Wang, MengWei Qu, Yu Bai, Roberto Soria, JiFeng Liu},

howpublished = {\url{https://github.com/Yu-Yang-Li/StarGLM}},

year = {2023}

}

Star History

Star History Chart