คู่มือทางเทคนิค

การเปิดใช้งาน SwiGLU และ Gated

SwiGLU เป็นฟังก์ชันการเปิดใช้งานแบบมีรั้วรอบขอบชิดซึ่งจะคูณการฉายภาพเชิงเส้นหนึ่งครั้งของอินพุตด้วยการฉายภาพครั้งที่สองที่เปิดใช้งาน Swish ซึ่งทำหน้าที่เป็นเกทที่เรียนรู้ได้และขึ้นอยู่กับข้อมูลภายในเลเยอร์ฟีดฟอร์เวิร์ดของหม้อแปลง

ภาพรวม

SwiGLU เป็นฟังก์ชันการเปิดใช้งานแบบมีรั้วรอบขอบชิดซึ่งจะคูณการฉายภาพเชิงเส้นหนึ่งครั้งของอินพุตด้วยการฉายภาพครั้งที่สองที่เปิดใช้งาน Swish ซึ่งทำหน้าที่เป็นเกทที่เรียนรู้ได้และขึ้นอยู่กับข้อมูลภายในเลเยอร์ฟีดฟอร์เวิร์ดของหม้อแปลง มันปรับปรุงคุณภาพโมเดลภาษาอย่างต่อเนื่อง ซึ่งเป็นเหตุผลว่าทำไม LLM สมัยใหม่เกือบทุกแห่งจึงใช้มัน

SwiGLU และ Gated Activations เป็นองค์ประกอบทางเทคนิคที่ส่งผลต่อคุณภาพของโมเดล ต้นทุนโครงสร้างพื้นฐาน เวลาแฝง และความน่าเชื่อถือในวงกว้าง

เจาะลึก

บล็อกฟีดฟอร์เวิร์ดของหม้อแปลงมาตรฐานคือเลเยอร์เชิงเส้นสองชั้นโดยมี ReLU หรือ GELU อยู่ระหว่างนั้น หน่วยเชิงเส้นแบบมีรั้วรอบขอบชิด เสนอโดย Dauphin และคณะ ในปี 2559 แบ่งการฉายภาพแรกออกเป็นสองซีก และใช้ครึ่งหนึ่งเพื่อเกตอีกครึ่งหนึ่งผ่านการคูณตามองค์ประกอบ SwiGLU ซึ่ง Noam Shazeer ได้รับความนิยมในปี 2020 ใช้ฟังก์ชัน Swish (SiLU) สำหรับเกตนั้น: output = (Swish(xW) * (xV)) W2 โดยมีเมทริกซ์น้ำหนัก 3 ตัวแทนที่จะเป็น 2 ตัว เกตช่วยให้เครือข่ายสามารถเลือกส่งผ่านหรือระงับข้อมูลต่อมิติได้ เนื่องจากการเพิ่มเมทริกซ์ตัวที่สามจะทำให้พารามิเตอร์เพิ่มขึ้น การใช้งานจะลดขนาดมิติที่ซ่อนอยู่เหลือประมาณสองในสาม ดังนั้นการประมวลผลทั้งหมดจึงเทียบได้กับ GELU MLP การทดลองของ Shazeer แสดงให้เห็นถึงความฉงนสนเท่ห์ที่วัดได้ และ LLaMA, PaLM และ Mistral ต่างก็นำมันมาใช้

ข้อมูลเชิงลึกทางเทคนิค

Swish คือ x * sigmoid(beta*x) ซึ่งเป็นฟังก์ชันที่ราบรื่นและไม่ซ้ำซาก ซึ่งต่างจาก ReLU ตรงที่ยอมให้ค่าลบเล็กๆ ผ่านได้ ใน SwiGLU Swish(xW) สาขา 'gate' จะสร้างค่าใกล้ 0 หรือ 1 ซึ่งจะคูณองค์ประกอบ xV ของสาขา 'value' ดังนั้นการมีส่วนร่วมของหน่วยที่ซ่อนอยู่แต่ละหน่วยจะถูกมอดูเลตโดยสัญญาณที่เรียนรู้และขึ้นอยู่กับอินพุต เมทริกซ์น้ำหนักตัวที่สามคือต้นทุน เคล็ดลับขนาดที่ซ่อนอยู่สองในสามช่วยให้งบประมาณ FLOP จับคู่กับเลเยอร์ส่งต่อวานิลลา

การเรียนรู้การเปิดใช้งาน SwiGLU และ Gated

SwiGLU เป็นฟังก์ชันการเปิดใช้งานแบบมีรั้วรอบขอบชิดซึ่งจะคูณการฉายภาพเชิงเส้นหนึ่งครั้งของอินพุตด้วยการฉายภาพครั้งที่สองที่เปิดใช้งาน Swish ซึ่งทำหน้าที่เป็นเกทที่เรียนรู้ได้และขึ้นอยู่กับข้อมูลภายในเลเยอร์ฟีดฟอร์เวิร์ดของหม้อแปลง มันปรับปรุงคุณภาพโมเดลภาษาอย่างต่อเนื่อง ซึ่งเป็นเหตุผลว่าทำไม LLM สมัยใหม่เกือบทุกแห่งจึงใช้มัน SwiGLU และ Gated Activations เป็นองค์ประกอบทางเทคนิคที่ส่งผลต่อคุณภาพของโมเดล ต้นทุนโครงสร้างพื้นฐาน เวลาแฝง และความน่าเชื่อถือในวงกว้าง หากต้องการสร้างความเข้าใจอย่างลึกซึ้ง ให้ถือว่า SwiGLU และ Gated Activations เป็นรูปแบบการดำเนินงาน ไม่ใช่คุณลักษณะเดียว: กำหนดผลลัพธ์ที่ต้องการ ชี้แจงสมมติฐาน และแยกสิ่งที่ระบบสามารถทำได้อย่างน่าเชื่อถือจากสิ่งที่ยังต้องใช้วิจารณญาณจากผู้เชี่ยวชาญ

ในทางปฏิบัติ ทีมที่แข็งแกร่งที่ใช้ SwiGLU และ Gated Activations จะปรับสถาปัตยกรรม ข้อมูล และตัวเลือกโครงสร้างพื้นฐานให้เหมาะสมโดยเทียบกับความน่าเชื่อถือและต้นทุน โดยจะบันทึกเกณฑ์ความสำเร็จที่ชัดเจน ทดสอบกับข้อมูลและขั้นตอนการทำงานที่สมจริง และทำซ้ำตามรูปแบบความล้มเหลวที่สังเกตได้ แทนที่จะชนะการวัดประสิทธิภาพเพียงครั้งเดียว นี่คือจุดที่ความเข้าใจทางทฤษฎีกลายเป็นความสามารถที่คงทนของผลิตภัณฑ์ นโยบาย และการดำเนินงาน

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี ในเวลาเดียวกัน การเพิ่มประสิทธิภาพเกณฑ์มาตรฐานหนึ่งรายการสามารถซ่อนจุดอ่อนของระบบในวงกว้างได้ แนวทางที่ยืดหยุ่นที่สุดคือการรวมความเร็วของการทดลองเข้ากับวินัยในการกำกับดูแล: ดำเนินการนำร่อง จับหลักฐาน เผยแพร่บันทึกการตัดสินใจ และอัปเดตการป้องกันอย่างต่อเนื่องเมื่อพฤติกรรมของโมเดล ความคาดหวังของผู้ใช้ และข้อกำหนดด้านกฎระเบียบมีการเปลี่ยนแปลง

ผลกระทบเชิงกลยุทธ์

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี

การตัดสินใจด้านสถาปัตยกรรมขับเคลื่อนประสิทธิภาพและต้นทุนการดำเนินงานเป็นเวลาหลายปี ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

การศึกษาด้านเทคนิคช่วยให้ทีมเลือกกลุ่มที่เหมาะสม ไม่ใช่แค่กลุ่มใหม่ล่าสุด

การศึกษาด้านเทคนิคช่วยให้ทีมเลือกกลุ่มที่เหมาะสม ไม่ใช่แค่กลุ่มใหม่ล่าสุด ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

ตัวเลือกทางวิศวกรรมที่ดีกว่าจะช่วยลดเหตุการณ์ด้านความน่าเชื่อถือในการผลิต

ตัวเลือกทางวิศวกรรมที่ดีกว่าจะช่วยลดเหตุการณ์ด้านความน่าเชื่อถือในการผลิต ในการปรับใช้คุณภาพสูง สิ่งนี้จะถูกแปลเป็นกฎการปฏิบัติงานที่วัดผลได้ ขอบเขตความเป็นเจ้าของ และขั้นตอนการตรวจสอบที่เกิดซ้ำ เพื่อให้ทีมสามารถปรับขนาดความมั่นใจแทนที่จะปรับขนาดความคลุมเครือ

อนาคตของการเปิดใช้งาน SwiGLU และ Gated

SwiGLU ได้รับการยึดที่มั่นเป็น MLP เริ่มต้นใน LLM แบบ open-weight และไม่น่าจะถูกแทนที่ในเร็วๆ นี้ คำแนะนำที่ใช้งาน ได้แก่ ตัวแปร GeGLU และ ReGLU, เคอร์เนล GPU ที่หลอมรวมซึ่งคำนวณทั้งสองการฉายภาพในการส่งผ่านครั้งเดียว และการรวม MLP ที่มีรั้วรอบขอบชิดกับผู้เชี่ยวชาญที่ผสมผสานกัน เพื่อให้ผู้เชี่ยวชาญแต่ละคนกลายเป็นบล็อก SwiGLU นักวิจัยยังกำลังศึกษาว่าเหตุใด gating จึงช่วยเพิ่มประสิทธิภาพ โดยมีเป้าหมายเพื่อออกแบบประตูที่มีราคาถูกลง

การใช้งานจริงในโลกแห่งความเป็นจริง

LLaMA, PaLM และ Mistral แทนที่เลเยอร์ฟีดไปข้างหน้าของ GELU ด้วย SwiGLU เพื่อลดความซับซ้อนในการคำนวณที่เท่ากัน

มิติที่ซ่อนอยู่จะถูกปรับขนาดเป็นประมาณสองในสาม (8/3 วัน) ดังนั้นเมทริกซ์เกตติ้งพิเศษจะไม่ทำให้ FLOP ขยายตัว

โมเดลแบบผสมผสานของผู้เชี่ยวชาญ เช่น Mixtral ใช้บล็อก SwiGLU เป็นเครือข่ายการส่งต่อฟีดต่อผู้เชี่ยวชาญ

วิชันซิสเต็มและหม้อแปลงหลายรูปแบบยืมเกท GeGLU/SwiGLU เพื่อปรับปรุงเลเยอร์ย่อย MLP

รูปแบบการดำเนินงาน

การเปิดใช้งาน SwiGLU และ Gated ในทางปฏิบัติ

LLaMA, PaLM และ Mistral แทนที่เลเยอร์ฟีดไปข้างหน้าของ GELU ด้วย SwiGLU เพื่อลดความซับซ้อนในการคำนวณที่เท่ากัน

LLaMA, PaLM และ Mistral แทนที่เลเยอร์ฟีดไปข้างหน้าของ GELU ด้วย SwiGLU เพื่อลดความยุ่งยากที่ลดลงด้วยการคำนวณที่เท่ากัน ทีมมักจะได้รับผลลัพธ์ที่ดีกว่าเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการทำงานที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

การเปิดใช้งาน SwiGLU และ Gated ในทางปฏิบัติ

มิติที่ซ่อนอยู่จะถูกปรับขนาดเป็นประมาณสองในสาม (8/3 วัน) ดังนั้นเมทริกซ์เกตติ้งพิเศษจะไม่ทำให้ FLOP ขยายตัว

มิติข้อมูลที่ซ่อนอยู่จะถูกปรับขนาดเป็นประมาณสองในสาม (8/3 วัน) ดังนั้นเมทริกซ์เกตติ้งพิเศษจะไม่ทำให้ FLOP ขยายตัว ปกติแล้วทีมจะได้รับผลลัพธ์ที่ดีกว่าเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการผลิตที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

การเปิดใช้งาน SwiGLU และ Gated ในทางปฏิบัติ

โมเดลแบบผสมผสานของผู้เชี่ยวชาญ เช่น Mixtral ใช้บล็อก SwiGLU เป็นเครือข่ายส่งต่อฟีดต่อผู้เชี่ยวชาญ

โมเดลแบบผสมผสานของผู้เชี่ยวชาญ เช่น Mixtral ใช้บล็อก SwiGLU เนื่องจากทีมเครือข่ายส่งต่อฟีดต่อผู้เชี่ยวชาญ มักจะได้รับผลลัพธ์ที่ดีกว่า เมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการทำงานที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

การเปิดใช้งาน SwiGLU และ Gated ในทางปฏิบัติ

หม้อแปลงวิชั่นและมัลติโมดัลยืมเกท GeGLU/SwiGLU เพื่อปรับปรุงเลเยอร์ย่อย MLP

Vision และหม้อแปลงหลายรูปแบบยืม GeGLU/SwiGLU gating เพื่อปรับปรุงเลเยอร์ย่อย MLP ของพวกเขา ทีมมักจะได้รับผลลัพธ์ที่ดีขึ้นเมื่อพวกเขากำหนดเกณฑ์คุณภาพล่วงหน้า รักษาเส้นทางการยกระดับโดยมนุษย์สำหรับกรณี Edge และติดตามทั้งประสิทธิภาพการทำงานที่เพิ่มขึ้นและต้นทุนข้อผิดพลาดเมื่อเวลาผ่านไป

ความเสี่ยงและรั้ว

!

การเพิ่มประสิทธิภาพเกณฑ์มาตรฐานหนึ่งรายการสามารถซ่อนจุดอ่อนของระบบในวงกว้างได้

!

ต้นทุนโครงสร้างพื้นฐานและการบำรุงรักษามักถูกประเมินต่ำไป

!

ช่องว่างด้านความปลอดภัยและความสามารถในการสังเกตสามารถเพิ่มขึ้นได้เมื่อระบบมีความซับซ้อนมากขึ้น

แผนงานการดำเนินงาน

1

กำหนดเป้าหมายเวลาแฝง คุณภาพ และต้นทุนก่อนนำไปใช้งาน

กำหนดเป้าหมายเวลาแฝง คุณภาพ และต้นทุนก่อนนำไปใช้งาน ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

2

เกณฑ์มาตรฐานภายใต้สภาวะโหลดและข้อมูลจริง

เกณฑ์มาตรฐานภายใต้สภาวะโหลดและข้อมูลจริง ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

3

การตรวจสอบเครื่องมือเพื่อหาข้อผิดพลาด การเบี่ยงเบน และผลกระทบต่อผู้ใช้

การตรวจสอบเครื่องมือเพื่อหาข้อผิดพลาด การเบี่ยงเบน และผลกระทบต่อผู้ใช้ ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

4

เตรียมเส้นทางการย้อนกลับและการตอบสนองต่อเหตุการณ์ก่อนปรับขนาด

เตรียมเส้นทางการย้อนกลับและการตอบสนองต่อเหตุการณ์ก่อนปรับขนาด ถือว่าแต่ละขั้นตอนเป็นเหมือนประตูหลักฐาน: หากไม่ตรงตามเกณฑ์ ให้หยุดการเปิดตัวชั่วคราว ปิดช่องว่าง จากนั้นจึงขยายการใช้งานเท่านั้น

สำรวจต่อไป